Heatmap only code. Also, adding some class diagrams.