Initial commit of heatmap generation code.