Added heatmap and treemap module