Minor change to main.cpp