Update Assignment_4.cpp