Added the ability to plot a heatmap of the datapoints in a dataset