Update to Homework 2