saving the predictions