diff --git a/src/plotbox/__init__.py b/src/plotbox/__init__.py
new file mode 100644
index 0000000..891dbd6
--- /dev/null
+++ b/src/plotbox/__init__.py
@@ -0,0 +1 @@
+from .plotbox import main
\ No newline at end of file
diff --git a/src/plotbox/__main__.py b/src/plotbox/__main__.py
new file mode 100644
index 0000000..2cffd2d
--- /dev/null
+++ b/src/plotbox/__main__.py
@@ -0,0 +1,4 @@
+from .plotbox import main()
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file
diff --git a/plotbox.py b/src/plotbox/plotbox.py
similarity index 97%
rename from plotbox.py
rename to src/plotbox/plotbox.py
index 98e21de..51529c1 100644
--- a/plotbox.py
+++ b/src/plotbox/plotbox.py
@@ -87,7 +87,4 @@ def main():
 
     do_plot(None)
 
-    plt.show()
-
-if __name__ == "__main__":
-    main()
\ No newline at end of file
+    plt.show()
\ No newline at end of file