almost done with hw1 part2