Add AI Class and more methods