diff --git a/.travis.yml b/.travis.yml
index ce134c5db0..dff413f6e8 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -33,3 +33,10 @@ script:
 
 after_success:
 - codecov
+
+deploy:
+  provider: pypi
+  username: "__token__"
+  password: "pypi-AgEIcHlwaS5vcmcCJGExMGU2ZDIwLWZjZjItNDE2OC1hMTAwLTBjMjcwODk0NmVhNgACOnsicGVybWlzc2lvbnMiOiB7InByb2plY3RzIjogWyJwcm9nbGVhcm4iXX0sICJ2ZXJzaW9uIjogMX0AAAYg4-9xmR2nprlOAV7X-Nv6Dkvi6N-WkTBEyTe8WvrWUC0"
+  on:
+    branch: main
diff --git a/proglearn/__init__.py b/proglearn/__init__.py
index a97a483d48..1b1275e534 100755
--- a/proglearn/__init__.py
+++ b/proglearn/__init__.py
@@ -2,3 +2,5 @@
 from .transformers import *
 from .voters import *
 from .deciders import *
+
+__version__ = "0.0.1"
diff --git a/setup.py b/setup.py
index 9aa056d51f..6488df4c32 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup, find_packages
+import os
 
 requirements = [
     "keras",
@@ -8,6 +9,12 @@
     "joblib",
 ]
 
+# Find mgc version.
+PROJECT_PATH = os.path.dirname(os.path.abspath(__file__))
+for line in open(os.path.join(PROJECT_PATH, "proglearn", "__init__.py")):
+    if line.startswith("__version__ = "):
+        VERSION = line.strip().split()[2][1:-1]
+
 with open("README.md", mode="r", encoding = "utf8") as f:
     LONG_DESCRIPTION = f.read()