added AI class
Added AI class