Part 1 of HW4 done