add one line to main.cpp