Refactored ai state info into AIStateController class.