updated scripts for augmented training set 2