updated CR to a new, shorter method.

cleaned diffusion.py a little

simply clean on vis