Update main.cpp

Fixed a minor annoyance in main.