Improve message styling