Create sample_generator.py

the code used to generate the sample data