Slight improvements to the legend