Added heatmap stuff.
removed extra stuff