Removed dependecy of state files on PEPS by switching to SimplePEPS copy of axes and tensordict, other small changes