Fixed main.cpp minor issue