Update FeatureExtractor.py