Improve multigrid notebook.