Finish project 2 problem 2