Accept kwarg for nlayers in plot_grid_dep