diff --git a/.gitignore b/.gitignore
index ecf5835..a779bdf 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,6 +1,7 @@
 # Ignore Python artifacts
 *.swp
 *.pyc
+/principalmapper.egg-info
 
 # Ignore image artifacts
 *.png
@@ -18,4 +19,4 @@
 /venv*
 
 # Ignore vscode
-.vscode/
\ No newline at end of file
+.vscode/
diff --git a/principalmapper/neo4j/__init__.py b/principalmapper/neo4j/__init__.py
new file mode 100644
index 0000000..ee4a1b0
--- /dev/null
+++ b/principalmapper/neo4j/__init__.py
@@ -0,0 +1,16 @@
+#  Copyright (c) NCC Group and Erik Steringer 2019. This file is part of Principal Mapper.
+#
+#      Principal Mapper is free software: you can redistribute it and/or modify
+#      it under the terms of the GNU Affero General Public License as published by
+#      the Free Software Foundation, either version 3 of the License, or
+#      (at your option) any later version.
+#
+#      Principal Mapper is distributed in the hope that it will be useful,
+#      but WITHOUT ANY WARRANTY; without even the implied warranty of
+#      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#      GNU Affero General Public License for more details.
+#
+#      You should have received a copy of the GNU Affero General Public License
+#      along with Principal Mapper.  If not, see <https://www.gnu.org/licenses/>.
+
+"""The Python module containing the code for identifying risks in an AWS account based on Principal Mapper data."""
diff --git a/setup.py b/setup.py
index ea072fa..dfd2271 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@
     packages=find_packages(exclude=("tests", )),
     package_data={},
     python_requires='>=3.5, <4',  # assume Python 4 will break
-    install_requires=['botocore', 'packaging', 'python-dateutil', 'pydot'],
+    install_requires=['botocore', 'packaging', 'python-dateutil', 'pydot', 'rich', 'neo4j'],
     entry_points={
         'console_scripts': [
             'pmapper = principalmapper.__main__:main'