modified chi calculations to the updated MDTraj codebase