modify part1 hw4