Made changes for part 3 of HW3