Updated UI styling