Small update in main.cpp.