Speed up on diffusion calcs