updated data.py