[TBD]Handling multi class classification in machine learning