Update main.cpp

Add some comments, added beginning prompt and number prompts.