Update main.cpp

Added user input