Improve prompt()