Finished up Problem 3