Inital commit of heatmap code. Needs work to make general.