think I fixed derivation of diffusion equation