Updated data.py