updates for Assignment2