Prediction model works (maybe)