diff --git a/__version__.py b/__version__.py
deleted file mode 100644
index 4609a4128406cca44479c45c79cacbea5d94badb..0000000000000000000000000000000000000000
--- a/__version__.py
+++ /dev/null
@@ -1,14 +0,0 @@
-# coding: utf-8
-
-"""
-TensorFlow implementation of the Lorentz Boost Network (LBN). https://arxiv.org/abs/1812.09722.
-"""
-
-
-__author__ = "Marcel Rieger"
-__copyright__ = "Copyright 2018-2019, Marcel Rieger"
-__license__ = "BSD"
-__credits__ = ["Martin Erdmann", "Erik Geiser", "Yannik Rath", "Marcel Rieger"]
-__contact__ = "https://git.rwth-aachen.de/3pia/lbn"
-__email__ = "marcel.rieger@cern.ch"
-__version__ = "1.0.1"
diff --git a/lbn.py b/lbn.py
index 9c866d03b920bdbc46a64425257e86a4d1c83a2b..99cebbf7048451601f5778e2fea521a357813bd9 100644
--- a/lbn.py
+++ b/lbn.py
@@ -5,6 +5,14 @@ TensorFlow implementation of the Lorentz Boost Network (LBN). https://arxiv.org/
 """
 
 
+__author__ = "Marcel Rieger"
+__copyright__ = "Copyright 2018-2019, Marcel Rieger"
+__license__ = "BSD"
+__credits__ = ["Martin Erdmann", "Erik Geiser", "Yannik Rath", "Marcel Rieger"]
+__contact__ = "https://git.rwth-aachen.de/3pia/lbn"
+__email__ = "marcel.rieger@cern.ch"
+__version__ = "1.0.1"
+
 __all__ = ["LBN", "FeatureFactoryBase", "FeatureFactory"]
 
 
diff --git a/requirements.txt b/requirements.txt
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0c8c3d2e7969d38f8b76ec8f79e2067fea30cdaa 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,2 @@
+numpy
+tensorflow
diff --git a/setup.py b/setup.py
index 361dcc0f349ac3c88fc325516bcdf54aa10ba4b8..e7909bb6d3eddee5bceda35c4d950e532ff1c883 100644
--- a/setup.py
+++ b/setup.py
@@ -34,20 +34,14 @@ with open(os.path.join(this_dir, "README.md"), "r") as f:
 with open(os.path.join(this_dir, "requirements.txt"), "r") as f:
     install_requires = [line.strip() for line in f.readlines() if line.strip()]
 
-# load package infos
-pkg = {}
-with open(os.path.join(this_dir, "__version__.py"), "r") as f:
-    exec(f.read(), pkg)
-
-
 setuptools.setup(
-    name="lbn",
-    version=pkg["__version__"],
-    author=pkg["__author__"],
-    author_email=pkg["__email__"],
-    description=pkg["__doc__"].strip().split("\n")[0].strip(),
-    license=pkg["__license__"],
-    url=pkg["__contact__"],
+    name=lbn.__name__,
+    version=lbn.__version__,
+    author=lbn.__author__,
+    author_email=lbn.__email__,
+    description=lbn.__doc__.strip().split("\n")[0].strip(),
+    license=lbn.__license__,
+    url=lbn.__contact__,
     keywords=keywords,
     classifiers=classifiers,
     long_description=long_description,
@@ -55,6 +49,6 @@ setuptools.setup(
     install_requires=install_requires,
     python_requires=">=2.7",
     zip_safe=False,
-    py_modules=["lbn"],
+    py_modules=[lbn.__name__],
     data_files=[(".", ["LICENSE", "requirements.txt", "README.md"])],
 )