Add a broom augment like function

[bailliem]

Add a function to return an augmented original dataset with the predictions and individual counterfactuals added (i..e predicted potential outcome). This would be similar or an extension of the broom::augment function.