adding class
fix