Add visualization for simple-grids