Update example for new data.py