Update main.cpp

restored...