Update heatmap.r

Made a few tweaks to the code and updated for new filename.