Heatmap js code up