update hw4 part1