Update for Assignment #5