Update homework 2