Problem 5 finished