Add #1 Python solution