Updated ai function's argument list