Modeling the α/ω thermal stability in shocked Zr: A coupling between dislocation removal and phase transformation