modified:   sample_generator.py