Misc UI enhancements