HW 4 part 2 - final