multi class classification python code