Python lab: Simplify create_example_data.py