Update main.cpp

Done