added heatmap and editing