Improve message display