Simple Python Leetcode Solution