HW4 - part 2