Adding the splitDataSet function, filtering rows