Added diffusion current as weight for diff coef