Minor adjustments.
Almost forgot to commit these files.
Heatmap.
Minor adjustments on UI.