Update problem.cpp