Complete the feature extractor module