AI now is working, left small fixes.