Update feature_extraction