added team removal logic for KO stage