Fixed AI System issues
Merge pull request #4 from GSP420/AI_fix

Fixed AI System issues