Use visual-grid-class instead of body

Allow user to define a class for the visual grid.