code to create heat map
initial