Added focus heatmap functionality