Add option to turn off the position wrapping for testing the diffusion coefficient
Add notebook to test the consistency of diffusion coefficient