problem 2 finished