Assignment 1, problem 3