Fixed the body_class method