Problem 2 finished