heatmap and attentionmap animation added