diff --git a/galsim_hub/__init__.py b/galsim_hub/__init__.py
index 255601d..78f234c 100644
--- a/galsim_hub/__init__.py
+++ b/galsim_hub/__init__.py
@@ -1,2 +1,11 @@
+from pkg_resources import DistributionNotFound
+from pkg_resources import get_distribution
+
+try:
+    __version__ = get_distribution(__name__).version
+except DistributionNotFound:
+    # package is not installed
+    pass
+
 from .input_generative import *
 from .generative_model import *
diff --git a/setup.py b/setup.py
index 2f84959..f587721 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,6 @@
 
 setup(
     name='galsim_hub',
-    version='0.0.1rc2',
     description='Support for Tensorflow Hub modules in GalSim',
     long_description=long_description,
     long_description_content_type='text/markdown',
@@ -23,11 +22,13 @@
         'Topic :: Scientific/Engineering :: Astronomy',
         # Pick your license as you wish
         'License :: OSI Approved :: MIT License',
-
-        'Programming Language :: Python :: 2.7',
         'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.6',
         'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
     ],
     packages=['galsim_hub'],
+    use_scm_version=True,
+    setup_requires=["setuptools_scm"],
     install_requires=['galsim', 'tensorflow_hub']
 )