HW4 part 2 v1.1