Improve the Prediction system