Fixes to make sure the scale shows up on plots

Continue onwards with the really hacky colour
scale that we use to mark out each of the
different types of expression value.