adding more ai functionality