update the part1 in hw5