changed the heatmap.js