Added old changes from commit cee58b78f13b79fd7c41cdeac3e3fecb570c2d06 .
Also the styling of the warning box over the heatmap