Update main.cpp

Correction to comment