diff --git a/03_make_figures.R b/03_make_figures.R
index c9591bf..e9f5ddc 100644
--- a/03_make_figures.R
+++ b/03_make_figures.R
@@ -9,6 +9,37 @@ ifelse(!dir.exists(file.path(getwd(), figure_dir)), dir.create(file.path(getwd()
 colors <- list(darkblue = "#062e57",
                lightblue = "#b1dcf3")
 
+# plot map
+plot_map <- function () {
+  plot(sf::st_geometry(extent_poly), 
+       col = colors$darkblue, 
+       extent = extent_bbox,
+       xlim = c(extent$longitude_min, extent$longitude_max),
+       ylim = c(extent$latitude_min, extent$latitude_max),
+       border = "black")
+  plot(sf::st_geometry(data$NHDWaterbody), 
+       col = colors$lightblue, 
+       border = NA,
+       extent = extent_bbox,
+       xlim = c(extent$longitude_min, extent$longitude_max),
+       ylim = c(extent$latitude_min, extent$latitude_max),
+       add = TRUE)
+  plot(sf::st_geometry(data$NHDArea), 
+       col = colors$lightblue, 
+       border = NA,
+       extent = extent_bbox,
+       xlim = c(extent$longitude_min, extent$longitude_max),
+       ylim = c(extent$latitude_min, extent$latitude_max),
+       add = TRUE)
+  plot(sf::st_geometry(data$NHDFlowline), 
+       col = colors$lightblue,
+       lwd = data$NHDFlowline$TotDASqKM^0.3204*0.0446,
+       border = NA,
+       extent = extent_bbox,
+       xlim = c(extent$longitude_min, extent$longitude_max),
+       ylim = c(extent$latitude_min, extent$latitude_max),
+       add = TRUE)
+}
 
 # save figure
 tiff(filename = paste0(figure_dir,"/map.tiff"),
@@ -17,43 +48,13 @@ tiff(filename = paste0(figure_dir,"/map.tiff"),
      units = "in",
      res = 600,
      compression = "lzw")
+plot_map()
+dev.off()
 
 png(filename = paste0(figure_dir, "/map.png"),
     width = 11,
     height = 17,
     units = "in",
     res = 600)
-
-# plot map
-
-plot(sf::st_geometry(extent_poly), 
-     col = colors$darkblue, 
-     extent = extent_bbox,
-     xlim = c(extent$longitude_min, extent$longitude_max),
-     ylim = c(extent$latitude_min, extent$latitude_max),
-     border = "black")
-plot(sf::st_geometry(data$NHDWaterbody), 
-     col = colors$lightblue, 
-     border = NA,
-     extent = extent_bbox,
-     xlim = c(extent$longitude_min, extent$longitude_max),
-     ylim = c(extent$latitude_min, extent$latitude_max),
-     add = TRUE)
-plot(sf::st_geometry(data$NHDArea), 
-     col = colors$lightblue, 
-     border = NA,
-     extent = extent_bbox,
-     xlim = c(extent$longitude_min, extent$longitude_max),
-     ylim = c(extent$latitude_min, extent$latitude_max),
-     add = TRUE)
-plot(sf::st_geometry(data$NHDFlowline), 
-     col = colors$lightblue,
-     lwd = data$NHDFlowline$TotDASqKM^0.3204*0.0446,
-     border = NA,
-     extent = extent_bbox,
-     xlim = c(extent$longitude_min, extent$longitude_max),
-     ylim = c(extent$latitude_min, extent$latitude_max),
-     add = TRUE)
-# finish saving figure
+plot_map()
 dev.off()
-dev.off()
\ No newline at end of file