Update problem2.cpp