Updated get_bsearch_heatmap