Update changes to adjust HW4 part 6
Update changes to adjust HW4 part 7