I'll use the same write_prediction is used in the other predictions.