Leetcode Two Sum Problem Python 3