Update 398hw01.cpp