Update main.cpp

Error fix