Create data.py