Add `class` to the safe attributes list