assignment 1 problem 4 solution