main.cpp modification