updated screen heatmap logic to fix bug