Update main.cpp

Fixed