hw4_part2 almost done.