Update for homework 2