@Override
    public void drawStroke(Graphics2D g) {
        g.setStroke(new BasicStroke(1f));
        if (highlighted) {
            g.setStroke(new BasicStroke(2f));
            g.setPaint(highlightedColor);
        } else if (selected) {
            g.setPaint(selectedColor);
        } else {
            g.setPaint(strokeColor);
        }

        g.draw(path);
    }