Fixed Small Error in the main.cpp