Adding AI class