Problem 1, Python