Updated prompt function

updated based on feedback from rox0r