main.cpp modify