Problem 1 Python