Refactor show heatmap on video

Add some variables.