Fix up prompt to my liking