add sample_data.py