Adding class update