fixed feature_extraction to allow different inputs
;