updated hw4 part1 #2