Update AiTest.scala