Improve current AI