Update for the 2nd homework.