small fixes

e.g. wrong labels in the heatmap image