modified data.py