modified Data.py