add test class AI