diff --git a/src/NODES.ipynb b/src/NODES.ipynb
index be0c14f..5191007 100644
--- a/src/NODES.ipynb
+++ b/src/NODES.ipynb
@@ -53,6 +53,8 @@
     "import copy \n",
     "from sklearn.model_selection import TimeSeriesSplit\n",
     "\n",
+
+#The class below initializes a feedforward neural network(with input layer, non-linear activation function, output layer) as well as a forward function with 2 inputs
     "class ODEFunc(nn.Module):\n",
     "    def __init__(self, dim):\n",
     "        super(ODEFunc, self).__init__()\n",