ENH: added functionality to plot heatmaps - this is useful for look at two dimensions of data at once