Fix 'tight_layout' warnings when running plotting

[kcormi]

matplotlib gives some warnings because the facet grid of heatmaps and 'tigh_layout' are not playing quite nicely together. Should make sure the plotting is working as expected on different platforms, and maybe suppress this error.