cleaned up the diffusion code.  mostly pep8 stuff.