make it work better with AI