Updating HW2 assignment