Update Problem_3.1.cpp