Update problemB.cpp