Updated UI and styles