Update data.py