Finishing the heat map integration