Added code to make a heatmap of neutrality index