change in main.cpp