Assignment 2 problem 1 done