Modify Builder.java: changes at DataSetSplitter initialization