minor changes in main.cpp