updated the classes