2 files changed, 5 insertions, 4 deletions

diff --git a/util/plot_dh.py b/util/plot_dh.py
index 33fc3eb..468e73a 100755
--- a/util/plot_dh.py
+++ b/util/plot_dh.py
@@ -104,4 +104,4 @@ if __name__ == "__main__":
     else:
         fig.set_size_inches(12, 10)
         ext = opts.output.name.split(".")[-1]
-        plt.savefig(opts.output, format=ext, dpi=400)
+        plt.savefig(opts.output, format=ext, dpi=400, bbox_inches='tight')
diff --git a/util/plot_gen.py b/util/plot_gen.py
index c37d7c2..98d8261 100755
--- a/util/plot_gen.py
+++ b/util/plot_gen.py
@@ -123,12 +123,13 @@ if __name__ == "__main__":
         axe_priv_hist.set_xlabel("private key Hamming weight")
         axe_priv_hist.set_ylabel("time ({})".format(unit))
         fig.colorbar(im, ax=axe_priv_hist)
-        
-    fig.text(0.01, 0.02, "Data size: {}".format(len(time_data)), size="small")
+
+    if plot_i > 2:
+        fig.text(0.01, 0.02, "Data size: {}".format(len(time_data)), size="small")
 
     if opts.output is None:
         plt.show()
     else:
         fig.set_size_inches(12, 10)
         ext = opts.output.name.split(".")[-1]
-        plt.savefig(opts.output, format=ext, dpi=400)
+        plt.savefig(opts.output, format=ext, dpi=400, bbox_inches='tight')