Update AIModule.js