minor changes to main.cpp file