change to main.cpp