Update problem1.cpp