Jung - zmiana koloru wierzchołka

Witam,

stworzyłem funkcję zmieniającą kolor wierzchołka w grafie. Problem polega na tym, że chciałbym mieć kilka wierzchołków w różnych kolorach a ta funkcja zamienia mi kolor obecnego wierzchołka na czerwony a następnie zmienia kolor nowego na zielony.

Jak zmodyfikować funkcję, aby przy okazji nie zmieniała mi wszystkich wierzchołków na czerwony?

Gdy dodam "return null" wierzchołek nie ma koloru. :)

Pozdrawiam :)

public void setVertexColor(City city) {
        Transformer<String, Paint> vertexPaint = new Transformer<String, Paint>() {
            @Override
            public Paint transform(String s) {
                if (city.getName().equals(s)) {
                    return Color.GREEN;
                } else {
                    return Color.RED;
                }
            }
        };
        visualizationViewer.getRenderContext().setVertexFillPaintTransformer(vertexPaint);
    }