Multi class classificatio part added