diff --git a/.gitignore b/.gitignore
index 9219d59..a1b8b5c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -49,4 +49,7 @@ MANIFEST
 *.spec
 
 # Sphinx documentation
-docs/_build/
\ No newline at end of file
+docs/**
+!docs/source/conf.py
+!docs/source/index.rst
+!docs/source/modules.rst
\ No newline at end of file
diff --git a/deepsocflow/__init__.py b/deepsocflow/__init__.py
index 1d1f13a..2db30dc 100644
--- a/deepsocflow/__init__.py
+++ b/deepsocflow/__init__.py
@@ -1 +1,2 @@
-from deepsocflow.hardware import Hardware, example_function
\ No newline at end of file
+from deepsocflow.py.hardware import *
+from deepsocflow.py.utils import *
\ No newline at end of file
diff --git a/deepsocflow/py/__init__.py b/deepsocflow/py/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/deepsocflow/hardware.py b/deepsocflow/py/hardware.py
similarity index 76%
rename from deepsocflow/hardware.py
rename to deepsocflow/py/hardware.py
index 1045ea0..e22f900 100644
--- a/deepsocflow/hardware.py
+++ b/deepsocflow/py/hardware.py
@@ -2,12 +2,13 @@
 import numpy as np
 from abc import ABC, abstractmethod
 import json
-from deepsocflow.utils import *
+from deepsocflow.py.utils import *
 
 
 class Hardware:
-    """_summary_
-    """    
+    """
+    Class to store static (pre-synthesis) parameters of the accelerator and export them to SystemVerilog and TCL scripts.
+    """
     def __init__(
             self, 
             processing_elements: (int, int) = (8,24), 
@@ -24,13 +25,27 @@ def __init__(
             weights_cache_kbytes: int =384, 
             edge_cache_kbytes: int|None = None
             ):
+        """
+        Args:
+            processing_elements (int, int, optional): _description_. Defaults to (8,24).
+            frequency_mhz (int, optional): _description_. Defaults to 250.
+            bits_input (int, optional): _description_. Defaults to 4.
+            bits_weights (int, optional): _description_. Defaults to 4.
+            bits_sum (int, optional): _description_. Defaults to 16.
+            bits_bias (int, optional): _description_. Defaults to 16.
+            max_batch_size (int, optional): _description_. Defaults to 512.
+            max_channels_in (int, optional): _description_. Defaults to 512.
+            max_channels_out (int, optional): _description_. Defaults to 512.
+            max_kernel_size (int, optional): _description_. Defaults to 13.
+            max_image_size (int, optional): _description_. Defaults to 32.
+            weights_cache_kbytes (int, optional): _description_. Defaults to 384.
+            edge_cache_kbytes (int | None, optional): _description_. Defaults to None.
+        """
         
         self.params = locals()
         self.params = {k:self.params[k] for k in self.params if not k == 'self'}
         
-        '''
-        Validation
-        '''
+        # Validation
         assert bits_input in [1,2,4,8] and bits_weights in [1,2,4,8]
         assert bits_bias  in [8,16,32]
         
@@ -46,16 +61,18 @@ def __init__(
         self.KH_MAX, self.KW_MAX = max_kernel_size if (type(max_kernel_size) == tuple) else (max_kernel_size, max_kernel_size)
         self.XH_MAX, self.XW_MAX = max_image_size if (type(max_image_size) == tuple) else (max_image_size, max_image_size)
 
+        self.RAM_WEIGHTS_DEPTH     = int((weights_cache_kbytes*1024)/(self.K_BITS*self.COLS*2))
         '''
-        Width of weights RAM   = K_BITS * COLS
-        Number of weights RAMs = 2
+        | Width of weights RAM   = K_BITS * COLS
+        | Number of weights RAMs = 2
         '''
-        self.RAM_WEIGHTS_DEPTH     = int((weights_cache_kbytes*1024)/(self.K_BITS*self.COLS*2))
 
+        self.RAM_EDGES_DEPTH       = edge_cache_kbytes if edge_cache_kbytes is not None else int(self.CI_MAX * self.XW_MAX * np.ceil(self.XH_MAX/self.ROWS)-1)
         '''
-        Depth of RAM needed for edge padding = k != 1 ? ci*xw*(blocks-1) : 0
+        | Depth of RAM needed for edge padding.
+        |     if k == 1 -> 0
+        |     else ci*xw*(blocks-1) 
         '''
-        self.RAM_EDGES_DEPTH       = edge_cache_kbytes if edge_cache_kbytes is not None else int(self.CI_MAX * self.XW_MAX * np.ceil(self.XH_MAX/self.ROWS)-1)
 
         self.L_MAX                 = int(np.ceil(self.XH_MAX//self.ROWS))
         self.CONFIG_BEATS          = 0
@@ -72,18 +89,29 @@ def __init__(
 
 
     def export_json(self, path='./hardware.json'):
+        '''
+        Exports the hardware parameters to a JSON file.
+        '''
+        
         with open(path, 'w') as f:
             json.dump(self.params, f, indent=4)
 
 
     @staticmethod
     def from_json(path='./hardware.json'):
+        '''
+        Creates the Hardware object from an exported JSON file.
+        '''
+        
         with open(path, 'r') as f:
             hw = Hardware(**json.load(f))
         return hw
 
 
     def export(self):
+        '''
+        Exports the hardware parameters to SystemVerilog and TCL scripts.
+        '''
 
         with open('rtl/include/config_hw.svh', 'w') as f:
             f.write(f'''
@@ -132,7 +160,3 @@ def export(self):
 set S_PIXELS_WIDTH_LF  {self.IN_BITS}
 set M_OUTPUT_WIDTH_LF  {self.OUT_BITS}
 ''')
-
-
-def example_function():
-    print("Hello World!")
\ No newline at end of file
diff --git a/deepsocflow/py/layers.py b/deepsocflow/py/layers.py
new file mode 100644
index 0000000..eb65747
--- /dev/null
+++ b/deepsocflow/py/layers.py
@@ -0,0 +1,20 @@
+from qkeras import QActivation
+from tensorflow.keras.layers import Input, Flatten, Add, MaxPooling2D
+import numpy as np
+
+# class QInput(Input):
+#     def __init__(self, shape, batch_size, hw, frac_bits, name=None):
+
+#         self.hw = hw
+#         self.input_frac_bits = input_frac_bits
+#         super().__init__(shape=shape, name=name)
+
+#         int_bits = hw.X_BITS - self.frac_bits + 1
+
+#         x = Input(shape=shape, batch_size=batch_size, name=name)
+#         x = QActivation(f'quantized_bits(bits={hw.X_BITS}, integer={int_bits}, False,True,1)')(x)
+
+#         return x
+
+
+
diff --git a/deepsocflow/py/model.py b/deepsocflow/py/model.py
new file mode 100644
index 0000000..68d93b4
--- /dev/null
+++ b/deepsocflow/py/model.py
@@ -0,0 +1,6 @@
+from qkeras import Model
+
+class QModel(Model):
+
+    def __init(self, inputs, outputs, name=None):
+        super().__init__(inputs, outputs, name=name)
\ No newline at end of file
diff --git a/deepsocflow/utils.py b/deepsocflow/py/utils.py
similarity index 100%
rename from deepsocflow/utils.py
rename to deepsocflow/py/utils.py
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 5811428..2480a85 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -17,7 +17,7 @@
 
 # -- Project information -----------------------------------------------------
 
-project = 'deepsocflow'
+project = 'deepsocflow.py'
 copyright = '2023, Abarajithan G, Zhenghua Ma'
 author = 'Abarajithan G, Zhenghua Ma'
 
@@ -36,6 +36,15 @@
     'sphinx.ext.viewcode',
     'sphinx.ext.napoleon',
 ]
+napoleon_google_docstring = True
+
+autodoc_default_options = {
+    'members': True,
+    'member-order': 'bysource',
+    'special-members': '__init__',
+    'undoc-members': False,
+    'exclude-members': '__weakref__'
+}
 
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ['_templates']
diff --git a/docs/source/deepsocflow.py.rst b/docs/source/deepsocflow.py.rst
new file mode 100644
index 0000000..2b4a11b
--- /dev/null
+++ b/docs/source/deepsocflow.py.rst
@@ -0,0 +1,45 @@
+deepsocflow.py package
+======================
+
+Submodules
+----------
+
+deepsocflow.py.hardware module
+------------------------------
+
+.. automodule:: deepsocflow.py.hardware
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+deepsocflow.py.layers module
+----------------------------
+
+.. automodule:: deepsocflow.py.layers
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+deepsocflow.py.model module
+---------------------------
+
+.. automodule:: deepsocflow.py.model
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+deepsocflow.py.utils module
+---------------------------
+
+.. automodule:: deepsocflow.py.utils
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: deepsocflow.py
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/deepsocflow.rst b/docs/source/deepsocflow.rst
deleted file mode 100644
index c4397d5..0000000
--- a/docs/source/deepsocflow.rst
+++ /dev/null
@@ -1,21 +0,0 @@
-deepsocflow package
-===================
-
-Submodules
-----------
-
-deepsocflow.hardware module
----------------------------
-
-.. automodule:: deepsocflow.hardware
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-Module contents
----------------
-
-.. automodule:: deepsocflow
-   :members:
-   :undoc-members:
-   :show-inheritance:
diff --git a/docs/source/index.rst b/docs/source/index.rst
index 3b5a200..d04d6ac 100644
--- a/docs/source/index.rst
+++ b/docs/source/index.rst
@@ -10,7 +10,7 @@ Welcome to deepsocflow's documentation!
    :maxdepth: 2
    :caption: Contents:
 
-   modules
+   deepsocflow.py
 
 Indices and tables
 ==================
diff --git a/docs/source/modules.rst b/docs/source/modules.rst
index 85f21e5..1ee8b7e 100644
--- a/docs/source/modules.rst
+++ b/docs/source/modules.rst
@@ -4,4 +4,4 @@ deepsocflow
 .. toctree::
    :maxdepth: 4
 
-   deepsocflow
+   deepsocflow.py