Added method to retrieve thermodynamic state as mdtraj object