2 files changed, 15 insertions, 10 deletions
diff --git a/wal b/pywal/__init__.py
index 942515f..6cf07e5 100755
--- a/wal
+++ b/pywal/__init__.py
@@ -1,4 +1,3 @@
-#!/usr/bin/env python3
 """
 wal - Generate and change colorschemes on the fly.
 Created by Dylan Araps.
@@ -12,9 +11,9 @@ import shutil
 import subprocess
 import sys
 
+__version__ = "0.1.6"
 
 # Internal variables.
-VERSION = "0.1"
 COLOR_COUNT = 16
 CACHE_DIR = pathlib.Path.home() / ".cache/wal/"
 
@@ -102,7 +101,7 @@ def process_args(args):
 
     # -v
     if args.v:
-        print(f"wal {VERSION}")
+        print(f"wal {__version__}")
         exit(0)
 
     # -i
@@ -572,6 +571,3 @@ def main():
     # This saves 10ms.
     # pylint: disable=W0212
     # os._exit(0)
-
-
-main()
diff --git a/setup.py b/setup.py
index 10c5307..0a3b76b 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,7 @@
 """wal - setup.py"""
 from setuptools import setup
 
+import pywal
 
 DESC = (
     "View the DOCS at: https://github.com/dylanaraps/pywal\n\n"
@@ -9,22 +10,30 @@ DESC = (
 )
 DESC = "".join(DESC)
 
+version = pywal.__version__
+download_url = f"https://github.com/dylanaraps/pywal/archive/{version}.tar.gz"
 
 setup(
     name="pywal",
-    version="0.1.6",
+    version=version,
     author="Dylan Araps",
     author_email="dylan.araps@gmail.com",
     description="🎨 Generate and change colorschemes on the fly",
     long_description=DESC,
     license="MIT",
     url="https://github.com/dylanaraps/pywal",
-    download_url="https://github.com/dylanaraps/pywal/archive/0.1.6.tar.gz",
-    scripts=["wal"],
+    download_url=download_url,
     classifiers=[
         "Environment :: X11 Applications",
         "License :: OSI Approved :: MIT License",
         "Operating System :: POSIX :: Linux",
         "Programming Language :: Python :: 3.6",
-    ]
+    ],
+    packages=['pywal'],
+    entry_points={
+        "console_scripts": [
+            "wal=pywal:main"
+        ]
+    },
+    python_requires=">=3.6"
 )