diff --git a/MANIFEST b/MANIFEST
new file mode 100644
index 0000000..b472c3d
--- /dev/null
+++ b/MANIFEST
@@ -0,0 +1,4 @@
+# file GENERATED by distutils, do NOT edit
+setup.cfg
+setup.py
+nnutils/__init__.py
diff --git a/nnutils/__init__.py b/nnutils/__init__.py
index e69de29..4fbc074 100644
--- a/nnutils/__init__.py
+++ b/nnutils/__init__.py
@@ -0,0 +1,2 @@
+from . import cnn_complexity_analyzer
+from . import training_pipeline
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 8e4c199..1cd3d27 100644
--- a/setup.py
+++ b/setup.py
@@ -1,17 +1,17 @@
 from distutils.core import setup
 
 setup(
-    name='nnutils',  # How you named your package folder (MyLib)
+    name='nn_pipeline',  # How you named your package folder (MyLib)
     packages=['nnutils'],  # Chose the same as "name"
-    version='0.1',  # Start with a small number and increase it with every change you make
+    version='0.2',  # Start with a small number and increase it with every change you make
     license='MIT',  # Chose a license from here: https://help.github.com/articles/licensing-a-repository
     description='Common utility functions for training and evaluating neural networks as well as analyzing computational complexity',
     # Give a short description about your library
     author='Rex Geng',  # Type in your name
     author_email='hgeng4@illinois.edu',  # Type in your E-Mail
-    url='https://github.com/rexxy-sasori/cnn_complexity_analyzer',
+    url='https://github.com/rexxy-sasori/nnutils.git',
     # Provide either the link to your github or to your website
-    download_url='https://github.com/rexxy-sasori/cnn_complexity_analyzer/archive/refs/tags/v0.1.tar.gz',  # I explain this later on
+    download_url='https://github.com/rexxy-sasori/nnutils/archive/refs/tags/v0.2.tar.gz',  # I explain this later on
     keywords=['Complexity', 'Profiler', 'CNN'],  # Keywords that define your package best
     install_requires=[  # I get to this in a second
         'torch>=1.8.1',