Updated UI code