Update Assignment 2.cpp