Update problem_4.cpp