added some classes, they work now