Fix function for adding the class to the body.