Wrote some heatmap code

should be pretty easy to adapt