Finished Problem 1 part 2