Added extract_feature UDF