Update main.cpp

fixed minor error at prompt