Implement from here https://arxiv.org/abs/2103.03098 in `report.py` make the color of the heatmap be defined by the effect size.