Update changes to adjust HW4 part 6