Изменения

color[index] = 2; // красит вершину в черный цвет

}