Update problem_1.cpp