Leetcode Problem Two Sum Python