Updated copy for HW4 part 2