diff --git a/.gitignore b/.gitignore
index 61449c6372ab9c0eaca775c34ea1b1b022bbcb77..423a52d16b06a08d3f9e81333eb2f10abb2542ab 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,11 +4,13 @@
 *.png
 !docs/logo/*.png
 *.svg
+!examples/model-comparison/reference/*.svg
 *.zip
 *.vtu
 *.vtp
 *.pvd
 *.pkl
+!examples/model-comparison/reference/*.pkl
 
 # Ignore __pycache__ directories
 */__pycache__/
diff --git a/src/bayesvalidrox/bayes_inference/bayes_inference.py b/src/bayesvalidrox/bayes_inference/bayes_inference.py
index cb7d871ec68c6d1f10b78da78ae1b4efdcb9060b..714d10a609be1add62d967f9fdf910e394592824 100644
--- a/src/bayesvalidrox/bayes_inference/bayes_inference.py
+++ b/src/bayesvalidrox/bayes_inference/bayes_inference.py
@@ -524,6 +524,7 @@ class BayesInference:
                 for xi in range(yi):
                     ax = axes[yi, xi]
                     ax.set_xlim(MetaModel.bound_tuples[xi])
+        plt.close()
 
         # Turn off gridlines
         for ax in figPosterior.axes:
diff --git a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
index 7ce17841ebb36bd64aae7be8c5fb799548886779..e054d23ef7a5b82289f68183828ed59c7a821c0e 100644
--- a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
+++ b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
@@ -36,6 +36,7 @@ class BayesModelComparison:
 
         # Compute total number of the measurement points
         MetaModel = list(modelDict.items())[0][1]
+        MetaModel.ModelObj.read_observation()
         self.n_meas = MetaModel.ModelObj.n_obs
 
         # Generate justifiability data
@@ -168,9 +169,10 @@ class BayesModelComparison:
                          ms=10, linewidth=2, label=model
                          )
 
-            plt.title(name.replace('_', '$-$'))
+            plt.title(f"Data generated by: {name.replace('_', '$-$')}")
             plt.ylabel("Weights")
             plt.xlabel("No. of measurement points")
+            ax.set_xticks(list(range(1, self.n_meas+1)))
             plt.legend(loc="best")
             fig.savefig(
                 f'{directory}modelWeights_{name}.svg', bbox_inches='tight'