Classify now returns string instead of an index