def get_potential_energy(self):
     return adiabatic(self.mol.get_positions())[self.now_state]      
 def get_potential_energy_multi(self):
     return adiabatic(self.mol.get_positions())