Added option to noisify training data