Create heatmap.py