* Extracted a class for holding the strings, used as labels + handled a
* corner case of classification