Fixed bug in AI
Updated AI