Update problem6.cpp