Updating figure 2 equations to have diffusion coefficients.