diff --git a/aimodel/src/lib/vis/embeddings.py b/aimodel/src/lib/vis/embeddings.py
index a0cb505..a0698b7 100644
--- a/aimodel/src/lib/vis/embeddings.py
+++ b/aimodel/src/lib/vis/embeddings.py
@@ -12,7 +12,7 @@ def vis_embeddings(filepath_output, features):
 	).fit(features)
 	
 	px = 1 / plt.rcParams['figure.dpi'] # matplotlib sizes are in inches :-( :-( :-(
-	width = 3840
+	width = 8000
 	height = 768
 	
 	plt.rc("font", size=20)
@@ -24,7 +24,7 @@ def vis_embeddings(filepath_output, features):
 	umap.plot.points(dimreducer,
 		ax=axes["A"]
 	)
-	plt.title(f"UMAP Dimensionality Reduction", fontsize=20)
+	axes["A"].set_title(f"UMAP Dimensionality Reduction", fontsize=20)
 	
 	# 2: Parallel coordinates
 	dataframe = pandas.DataFrame(features)
@@ -39,7 +39,7 @@ def vis_embeddings(filepath_output, features):
 		sort_labels=True
 	)
 	
-	plt.title(f"Parallel coordinates plot", fontsize=20)
+	axes["B"].set_title(f"Parallel coordinates plot", fontsize=20)
 	
 	plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", fontsize=28, weight="bold")
 	plt.savefig(filepath_output)
\ No newline at end of file