Improve prompt rendering