-adjusted the two diffusion constant calculation function docstrings to specify that input MSD values are in A ** 2 rather than A