edit main.cpp