Cleaned up diffusion code
Made diffusion test more readable
Write output with coords
Layout the validation output better