assignment 1 problem 2 solved