Refactor css_classes