Continue to update for Assignment 3