Create feature_extraction.py