Improved legibility of heatmap plots

Especially for the legend. In addition, better handling of plots with
large number of rows/columns.