Update prob5/prob5.cpp