Working on LeetCode with Python