MAINT: updated grid example for new plot_colorbar syntax