IMprovements in Heatmap.js