Edit heatmap code to use built-in numpy 2D histogram function.