Assignment 1, problem 5