changed dataset generator to generate more values