problem 4 finished