Update to Assignment 3