AI Prediction Module Introduced