Update problem001.cpp