Refactor Augmentation module to inherit a ```AugmentationMixIn``` that will handle the deterministic/stochastic nature of the transformers