Part 2 of Assignment 1