Update prog1.cpp