Leetcode Two Sum Python 3