Added diffusion tests