Leetcode Simple Python Program