AI Feature Improvements