Edit main.cpp