Update main.cpp

Code readability/style update.