Added comments throughout, added rough heatmap layout