Update changes to adjust HW4 assignment2