Tweak key-drawing code