Update prog2.cc