diff --git a/opencl_fdtd/__init__.py b/opencl_fdtd/__init__.py
index e3e95b4..b621c19 100644
--- a/opencl_fdtd/__init__.py
+++ b/opencl_fdtd/__init__.py
@@ -1 +1,5 @@
 from .simulation import Simulation, type_to_C
+
+__author__ = 'Jan Petykiewicz'
+
+version = '0.4'
diff --git a/setup.py b/setup.py
index c7e55f3..8b098db 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,10 @@
 #!/usr/bin/env python3
 
 from setuptools import setup, find_packages
+import opencl_fdtd
 
 setup(name='opencl_fdtd',
-      version='0.4',
+      version=opencl_fdtd.version,
       description='OpenCL FDTD solver',
       author='Jan Petykiewicz',
       author_email='anewusername@gmail.com',