See #29 . Using the template in `musings/example-recipe.R`, add a method for `predict` so that if `new_data` has class `epi_df`, we keep the keys in the prediction.