Updated UI Code