changes in implementation of diffusion approximation, still to work