Update assignment10.cpp