Update class on save