Hi @wietsedv, Thanks for your great work! Could you please point me where is the code for drawing the heatmap in you paper? Best, Chiyu