add test code to generate heatmap & registration