Slight modification in main.cpp.