Update main.cpp

try 2