Remove debug statements from data.py