Added ai however results are inconsistent