Update main4.cpp