Update main.cpp

main updated