Update to Homework 1