Update to Assignment 4