Added more methods to update class