Assignment 2 Probl 1 done