diff --git a/mode_converter.ipynb b/mode_converter.ipynb
index f57aab2..be0a993 100644
--- a/mode_converter.ipynb
+++ b/mode_converter.ipynb
@@ -197,7 +197,7 @@
     "]\n",
     "#mpa.Near2FarFields return a numpy array with shape (num_of_points,nfreq,6) where the \n",
     "#third axis are the field components Ex,Ey,Ez,Hx,Hy,Hz\n",
-    "FarFields = mpa.Near2FarFields(sim,NearRegions,far_z)\n",
+    "FarFields = mpa.Near2FarFields(sim, NearRegions, far_z)\n",
     "ob_list = [FarFields]\n",
     "\n",
     "#this the objective to function\n",
@@ -205,7 +205,7 @@
     "def J1(FF):\n",
     "    #in this case we want to maximize field intensity\n",
     "    # at the position of the focal spot \n",
-    "    obj_to_return = npa.mean(npa.abs(FF[0,:,1]) ** 2)\n",
+    "    obj_to_return = npa.mean(npa.abs(FF[0,:,0]) ** 2)\n",
     "    return obj_to_return\n",
     "#returns the first point of all frequencies of the Ey component"
    ]
@@ -258,7 +258,8 @@
     ")\n",
     "\n",
     "######CHECK PLOT######\n",
-    "figure = opt.plot2D(True, output_plane=output_plane)\n"
+    "figure = opt.plot2D(True, output_plane=output_plane)\n",
+    "plt.savefig(\"results/simulation_setup.pdf\")"
    ]
   },
   {
diff --git a/mode_converter.py b/mode_converter.py
index 5d6289d..11090c5 100644
--- a/mode_converter.py
+++ b/mode_converter.py
@@ -142,7 +142,7 @@ NearRegions = [
 ]
 #mpa.Near2FarFields return a numpy array with shape (num_of_points,nfreq,6) where the 
 #third axis are the field components Ex,Ey,Ez,Hx,Hy,Hz
-FarFields = mpa.Near2FarFields(sim,NearRegions,far_z)
+FarFields = mpa.Near2FarFields(sim, NearRegions, far_z)
 ob_list = [FarFields]
 
 #this the objective to function
@@ -150,7 +150,7 @@ ob_list = [FarFields]
 def J1(FF):
     #in this case we want to maximize field intensity
     # at the position of the focal spot 
-    obj_to_return = npa.mean(npa.abs(FF[0,:,1]) ** 2)
+    obj_to_return = npa.mean(npa.abs(FF[0,:,0]) ** 2)
     return obj_to_return
 #returns the first point of all frequencies of the Ey component
 
@@ -190,7 +190,7 @@ opt = mpa.OptimizationProblem(
 
 ######CHECK PLOT######
 figure = opt.plot2D(True, output_plane=output_plane)
-
+plt.savefig("results/simulation_setup.pdf")
 
 # %%
 def mapping(x,eta,beta):