diff --git a/docs/reference/node_reference.md b/docs/reference/node_reference.md
index bdf645c..a85367a 100644
--- a/docs/reference/node_reference.md
+++ b/docs/reference/node_reference.md
@@ -155,7 +155,6 @@
 ## PyTorch wrapper - Training
 | Node | Description |
 | --- | --- |
-| [Instantiates the most basic optimizer to update W using the below formula](pto_simple.md) | $$ |
 | [Pt Apply Function](pt_apply_function.md) | Applies a function to the input tensor. |
 | [Pt Data Loader](pt_data_loader.md) | Loads data from a dataset node and creates a PyTorch DataLoader.   |
 | [Pt Data Loader From Tensors](pt_data_loader_from_tensors.md) | Creates a Torchvision Dataloader from a pair of tensors. |
@@ -175,6 +174,7 @@
 | [Pto Lr Scheduler Cosine Annealing](pto_lr_scheduler_cosine_annealing.md) | Creates a cosine annealing learning rate scheduler for an optimizer. |
 | [Pto Lr Scheduler Reduce On Plateau](pto_lr_scheduler_reduce_on_plateau.md) | Creates a reduce-on-plateau learning rate scheduler for an optimizer. |
 | [Pto Lr Scheduler Step](pto_lr_scheduler_step.md) | Creates a StepLR learning rate scheduler for an optimizer. |
+| [Pto Simple](pto_simple.md) |     Instantiates the most basic optimizer to update W using the below formula: |
 | [Ptv Dataset](ptv_dataset.md) | A Torchvision Dataset class wrapper. |
 | [Ptv Dataset Len](ptv_dataset_len.md) | A wrapper class that calls Python len on a dataset. |
 | [Ptv Dataset Loader](ptv_dataset_loader.md) | A node to combine the dataset and data loader into a single node. |
diff --git a/docs/reference/pto_simple.md b/docs/reference/pto_simple.md
index 264f9f1..71852d1 100644
--- a/docs/reference/pto_simple.md
+++ b/docs/reference/pto_simple.md
@@ -1,4 +1,5 @@
-# Instantiates the most basic optimizer to update W using the below formula
+# Pto Simple
+Instantiates the most basic optimizer to update W using the below formula:
 $$
 W -= lr * G
 $$
diff --git a/modules/pytorch_wrapper/pto_simple.py b/modules/pytorch_wrapper/pto_simple.py
index 93e8eb6..ba7da8e 100644
--- a/modules/pytorch_wrapper/pto_simple.py
+++ b/modules/pytorch_wrapper/pto_simple.py
@@ -46,6 +46,7 @@ $$
 
 class PtoSimple:
     """
+    Pto Simple:
     Instantiates the most basic optimizer to update W using the below formula:
 $$
 W -= lr * G