start refactoring data set generation for benchmark