Update to Assignment 2