Update Assignment-1.cpp