Added ai function decleration