Added UI enhancements and styling