HW3 part 2 done