improve grid visualizer