Finished Problem 1 of Assignment 2