Assignment 1, problem 4