Added AI class w/ tests