diff --git a/README.md b/README.md
index 7f30791..fa9716f 100644
--- a/README.md
+++ b/README.md
@@ -5,25 +5,11 @@ ndx-franklab-novela is a python package containing NWB custom extensions.
 
 # Extensions
 
-### Edge
-An undirected edge connecting two nodes in a graph. <br>
+### AssociatedFiles
+Representation of associated files in NWB <br>
 **Attributes:** <br>
-**name**  `string`: name of this segement node <br>
-**edge_nodes**  `array_data`, `data`: the names of the two nodes in this undirected edge' <br>
-
-#### Node
-A generic graph node. Subclass for more specific types of nodes. <br>
-**Attributes:** <br>
-    **name**  `string`: name of this node <br>
-    **value**  `int`: value of this node' <br>
-
-#### Apparatus
-Topological graph representing connected components of a behavioral apparatus. <br>
-**Attributes:** <br>
-    **name**  `string`: name of apparatus <br>
-    **nodes**  `list`: Node objects contained in this apparatus <br>
-    **edges**  `list`: Edge objects contained in this apparatus <br>
-
+    **description**  `string`: description of associated file <br>
+    **content**  `string`: content of associated file <br>
 
 #### HeaderDevice
 Representation of HeaderDevice in NWB. <br>
@@ -49,14 +35,20 @@ Representation of HeaderDevice in NWB. <br>
     **system_time_at_creation**  `string`: system_time_at_creation from header global configuration <br>
     **file_path**  `string`: file_path from header global configuration <br>
 
-#### NTrode
-Representation of NTrode object in NWB <br>
+#### ShanksElectrode
+Representation of ShanksElectrode object in NWB <br>
+**Attributes:** <br>
+    **name**  `string`: name of the shank <br>
+    **rel_x**  `float`: the rel_x value of this electrode <br>
+    **rel_y**  `float`: the rel_y value of this electrode <br>
+    **rel_z**  `float`: the rel_z value of this electrode of p
+
+#### Shank
+Representation of Shank object in NWB <br>
 **Attributes:** <br>
-    **ntrode_id**  `int`: id of electrode group <br>
-    **electrode_group_id**  `int`: id of electrode group<br>
-    **bad_channels**  `array_data`: ids of bad channel <br>
-    **map**  `array_data`: map of ntrode <br>
-    
+    **name**  `string`: name of the shank <br>
+    **shanks_electrodes**  `list`,`tuple`:  'electrodes in shank', 'default': list() <br>
+
 #### Probe
 Representation of Probe object in NWB <br>
 **Attributes:** <br>
diff --git a/src/pynwb/ndx_franklab_novela/associated_files.py b/src/pynwb/ndx_franklab_novela/associated_files.py
index 21b5acb..bf900b9 100644
--- a/src/pynwb/ndx_franklab_novela/associated_files.py
+++ b/src/pynwb/ndx_franklab_novela/associated_files.py
@@ -6,6 +6,7 @@
 @register_class('AssociatedFiles', 'ndx-franklab-novela')
 class AssociatedFiles(NWBDataInterface):
     """ Representation of associated files in NWB """
+
     __nwbfields__ = ('description', 'content')
 
     @docval(*get_docval(NWBDataInterface.__init__) + (
diff --git a/src/pynwb/ndx_franklab_novela/header_device.py b/src/pynwb/ndx_franklab_novela/header_device.py
index 12c009d..fb338f2 100644
--- a/src/pynwb/ndx_franklab_novela/header_device.py
+++ b/src/pynwb/ndx_franklab_novela/header_device.py
@@ -6,6 +6,7 @@
 @register_class('HeaderDevice', 'ndx-franklab-novela')
 class HeaderDevice(Device):
     """Configuration comes from SpikeGadgets recording system. Represented as HeaderDevice in NWB"""
+
     __nwbfields__ = ('headstage_serial', 'headstage_smart_ref_on', 'realtime_mode', 'headstage_auto_settle_on',
                      'timestamp_at_creation', 'conntroller_firmware_version', 'controller_serial',
                      'save_displayed_chan_only', 'headstage_firmwavare_version', 'qt_version', 'compile_date',