added save_aug flag to mnist_augmented to choose whether or not to save
augmented datasets.
corrections on fast test script