diff --git a/notebooks/Coronagraph_Basics.ipynb b/notebooks/Coronagraph_Basics.ipynb
index 5294bdfb..a23633d5 100644
--- a/notebooks/Coronagraph_Basics.ipynb
+++ b/notebooks/Coronagraph_Basics.ipynb
@@ -496,7 +496,7 @@
     }
    ],
    "source": [
-    "from pynrc.obs_nircam import plot_hdulist\n",
+    "from pynrc.nb_funcs import plot_hdulist\n",
     "from matplotlib.patches import Circle\n",
     "\n",
     "fig, axes = plt.subplots(1,3, figsize=(14,4.3))\n",
@@ -594,7 +594,7 @@
     }
    ],
    "source": [
-    "from pynrc.obs_nircam import plot_contrasts, plot_planet_patches, plot_contrasts_mjup\n",
+    "from pynrc.nb_funcs import plot_contrasts, plot_planet_patches, plot_contrasts_mjup\n",
     "import matplotlib.patches as mpatches\n",
     "\n",
     "# fig, ax = plt.subplots(figsize=(8,5))\n",
diff --git a/pynrc/pynrc_core.py b/pynrc/pynrc_core.py
index 0136d191..d85ac3f1 100644
--- a/pynrc/pynrc_core.py
+++ b/pynrc/pynrc_core.py
@@ -1567,7 +1567,7 @@ def bg_zodi(self, zfact=None, **kwargs):
 
         return fzodi_pix
         
-    def saturation_levels(self, sp, full_size=True, ngroup=2, **kwargs):
+    def saturation_levels(self, sp, image = None, full_size=True, ngroup=2, **kwargs):
         """Saturation levels.
         
         Create image showing level of saturation for each pixel.