Improved UI styling