Fixed some bugs in the AI