[dev] add multi-class support

[nickvazz]

• [ ] import other classes
• [ ] stack each class as its own layer in the preprocessing step