Organized AI code into more classes