Ait model integration v2