changed it to shuffle training data