Update main.cpp

Simple style changes