Improve prompt formatting