added hw4 problem 1