updated the class