Touch up main.cpp.