Piano keyboard should take track colours from the mixer

In the mixer it is possible to change an instrument's "Track Colour". My suggestion is that during playback with the piano keyboard open, the notes that appear when played by an instrument should match this track colour, rather than all being blue. Also, an option to make certain tracks such as drums not appear on the keyboard at all. As it stands with multiple instruments the keyboard can be a bit of a mess, but it has the potential to be an interesting visual aid with only these minor changes.