Migrate gmx msd to the trajectoryanalysis framework