HW1 part 4 done