Worked on Assignment 4: Problem 1