Update assignment1.cpp