Edited some bug in FeatureExtractor