Adding update class