plot_grid function

Plot the grid cell classification