Update for part 2 of HW4