Update problem_5.cpp