Update Assignment01.cpp