Heatmaps working. Fixed bug with output folder not being correct