Add ai but it definitely not work now