Update main.cpp

New version