Improve prompt display