Update hw4 part1 4