enhance ai module
Merge pull request #37 from PeterDaveHello/enhance_ai

[ai] enhance ai module