added more to heatmap.js