Fixing previous commit for main.cpp.