removed extra logging and implemented custom Augmenter class