Update hw4 part1 2