add multiclass support
remove unused cells/code