Refactoring + ability to save training set to arff file