added correlation + diffusion functions