makes AI class more modular