one more correction in main.cpp