Clean output
Merge back missing elements on unsupervised learning + recompute plots