Added heatmap legend. fixes #620.