Update feature_extractor.py