add_prediction function

Add predictions to a data frame