heatmap fixes

#11