wip mdtraj integration 2