changing colors note stem

Hi all,

I'm using te colornotes plugin to change the colors of the notes, but I would also like to change the colors of the stem and hooks. I tried to alter the code by copying the "note.accidental" part and changing it to "note.stem":

        if (note.accidental) {
              if (note.accidental.color == black)
                    note.accidental.color = colors[note.pitch % 12];
              else
                    note.accidental.color = black;
        }

        if (note.stem) {
              if (note.stem.color == black)
                    note.stem.color = colors[note.pitch % 12];
              else
                    note.stem.color = black;
        }

But this doesn't work.

What am I doing wrong?