HW4 part 2
HW4 part 2 fix img