Add heatmap tracking