Updating the class again