Added diffusivity subroutine