Finish problem 2 of problem set 5