Update prog.cpp