Update main.cpp

correction