First python leetcode