Update main.cpp

Improve if else