Update main.cpp

Corrected a little error.