Updating Assignment 4