diff --git a/setup.py b/setup.py
index 05d2b43..33c69ad 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
     name = 'hiddenlayer',
     # packages = ['hiddenlayer'],
     packages = setuptools.find_packages(),
-    version = '0.1',
+    version = '0.2',
     license="MIT",
     description = 'Neural network graphs and training metrics for PyTorch and TensorFlow',
     author = 'Waleed Abdulla <waleed.abdulla@gmail.com>, Phil Ferriere <pferriere@hotmail.com>',