Added multiclass classification, tested on wine dataset