Add heatmap figure (to be moved to libomx). Updated, improved notebooks.