Added 'toggle ai' button