diff --git a/main_torch.ipynb b/main_torch.ipynb
index 0995ae9..8987333 100644
--- a/main_torch.ipynb
+++ b/main_torch.ipynb
@@ -2390,7 +2390,7 @@
     }
    ],
    "source": [
-    "for i in range(0, 4):\n",
+    "for i in range(0, 4): #plot both graphs\n",
     "    print(\"GRU Evaluation for Cell {}\".format(i + 5))\n",
     "    gru_outputs, targets, gru_MAE, gru_RMSE = evaluate(gru_model, test_x[i], test_y[i], label_sc)\n",
     "    plt.figure(figsize=(14, 10))\n",