Python project experimenting with heatmaps