modifications to feature extractor