Problem 2 complete