Updated AITestUtility with more stat tracking.