Update Assignment 6