Changes for part 3 of HW3