Problem 2
Finished