UI improvements for messages