Fixed bugs in FeatureExtraction class