CLN: clean up data.py