Ability to pick classes to be used in supervised classification

[Green-Cat]

Currently only the number of classes used in the classification can be picked, would be better to let users also individually pick the classes they would like to see used.

Depends on #1.