wip mdtraj integration