Update problem3.cpp