Python module to create heatmaps