Update hw4 part1 5