Minor UI enhancements