Fixed import error in data.py