added code for transit heatmap