Update prog5.04.cpp