Added prompt in main