updating matrix diffusion solution in line with modifications to manuscript