added AI class (relates to #2)