Problem 1 finished