diff --git a/Inst_eddies/Analysis/distribution_function.py b/Inst_eddies/Analysis/distribution_function.py
index 400d40921b1894940a3aa896883e3e8282f047f9..2f661173ca2d19a8d570b8b2de3aef2b9754767d 100755
--- a/Inst_eddies/Analysis/distribution_function.py
+++ b/Inst_eddies/Analysis/distribution_function.py
@@ -136,7 +136,15 @@ if __name__ == "__main__":
     parser.add_argument(
         "dir", help="directory containing collection of shapefiles"
     )
+    parser.add_argument("--save", action="store_true")
     args = parser.parse_args()
     d = read(args.dir)
     fig_list = plot_all([d])
-    plt.show()
+
+    if args.save:
+        for i, fig in enumerate(fig_list):
+            fig.savefig(f"figure_{i}.png")
+
+        print(f"Created figure_[0-{len(fig_list)-1}].png")
+    else:
+        plt.show()