Added template option and heatmap template