adds visualization, heatmap dummy #matplotlib
buggy heatmap poc
fixes heatmap rendering