Update main.cpp

More changes