added a way to save new test and train files