Tweak AI logic