Working on Assignment 2 Problem 1