python for leetcode