update hw2 part2