Add AI functionality