Modify main.cpp file