Updated UI design