I think the Cataphract should be a regional unit available in the Stable replacing the Knight line for Byzantines and Armenians.
Byzantines keep their UT to buff them and get a new UU that is the land version of the Fireship.
Armenian Cataphracts could get 30HP from their UT since they are missing the last armour upgrade.