added heatmap functionality with weight and with weight simulation