Reverted back to older version of feature extraction script