working on the classes