Added Ai class