Improved AI integration