Correcting the diffusion approximation at the boundary.

Correcting the diffusion approximation at the boundary.