Added new variable for the visual-grid

Allow user to define the visual grid class.