Added AI classes