Improving multigrid for 3-D electro-magnetic diffusion on stretched grids