Updated UI based on assignment