Adding class to callback