Prediction workinggad  Need to fix off-by-one labeling again. Will do tomorrow.