Background Colors not updating

This code is exactly what I needed for a project, so thank you in advance of my question. 

I am unable to update the heatmap to account for the new facecolors ... the map is rendered using only the cmap colors. Is there a work around for this?