modify feature_extraction.py