small change to feature_extraction.py