HeatMap UI corrected