Create data_extraction.py