Update main.cpp

Main cpp file
.