improve multigrid support