for hw4 part2 #2