Making AI function properly...