HW4 - end of part 1