part 2 hw4 almost finished