Began work on diffusion step and associated tests.