made small changes in heatmap.py