Assignment 1 problem 3