Update to Assignment 1