UI enhancements for messages