Add body_class action and associated function