Finished Problem 2