Updated badge UI