Refactored CPU AI to separate class