Solve #2 [python]