Add support for multi classification
remove warnings