added code to generate a test dataset