enhance ai module #2
Merge pull request #38 from PeterDaveHello/enhance_ai

enhance ai module #2