Update main.cpp

Just needed to copy something. Nothing big