Just tested some new files
some adjustments to the diffusion plot, and update on notebook