sw_augment function

Augment data according to a tidied model