Update assignment.cpp