diff --git a/.gitignore b/.gitignore
index af15276..7598767 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 drawitor/__pycache__/
+dist/
diff --git a/drawitor/__init__.py b/drawitor/__init__.py
index 4306520..ce7e070 100644
--- a/drawitor/__init__.py
+++ b/drawitor/__init__.py
@@ -1,5 +1,7 @@
 # Exports
 from .core import draw, draw_image, draw_gif
 
+from .cli import main
+
 
 __version__ = "0.1.0"
diff --git a/drawitor/__main__.py b/drawitor/__main__.py
index 122ccb9..8273c4f 100644
--- a/drawitor/__main__.py
+++ b/drawitor/__main__.py
@@ -1,10 +1,3 @@
-from argparse import ArgumentParser
-from .core import draw
+from . import main
 
-drawitor = ArgumentParser(description="Draw Images or GIFs in your terminal")
-
-drawitor.add_argument("image", type=str, help="Image to draw")
-
-args = drawitor.parse_args()
-
-draw(args.image)
+main()
diff --git a/drawitor/cli.py b/drawitor/cli.py
new file mode 100644
index 0000000..b6ee7d7
--- /dev/null
+++ b/drawitor/cli.py
@@ -0,0 +1,12 @@
+from argparse import ArgumentParser
+from .core import draw
+
+
+def main():
+    drawitor = ArgumentParser(description="Draw Images or GIFs in your terminal")
+
+    drawitor.add_argument("image", type=str, help="Image to draw")
+
+    args = drawitor.parse_args()
+
+    draw(args.image)
diff --git a/pyproject.toml b/pyproject.toml
index 3295183..0b30de2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "drawitor"
-version = "0.1.0"
+version = "0.1.1"
 description = "Draw Images or GIFs in your terminal."
 authors = ["Eliaz Bobadilla <eliaz.bobadilladev@gmail.com>"]
 readme = "README.md"