We should add an option to present result as a heatmap. I think it could be a good idea to use matplotlib's pyplot package.

Matplotlib’s flexibility suits publication-quality charts that require precise adjustments and annotations. Seaborn speeds up creating heatmaps, regression plots, and pairplots with minimal coding.