Updating Assignment 6