Added mathematica notebook and diffusion plot