Improved prompt method. Moved exit strategy to running program
working Week 3