Added comments to code
Optimized Heatmap code. The function now normalizes the values by itself to a range between 0 and 1