diff --git a/.gitignore b/.gitignore
index ca01579..dfee8ec 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,7 +7,6 @@
 checkpoint
 logs/*
 tf_logs/*
-images/**/*.png
 images/**/*.dot
 my_*
 datasets/flowers
diff --git a/02_end_to_end_machine_learning_project.ipynb b/02_end_to_end_machine_learning_project.ipynb
index 295931d..1f1375e 100644
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -1052,7 +1052,8 @@
     }
    ],
    "source": [
-    "housing[\"income_cat\"].hist()"
+    "housing[\"income_cat\"].hist()\n",
+    "save_fig('income_category_hist')"
    ]
   },
   {
@@ -7743,7 +7744,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.5"
+   "version": "3.6.5"
   },
   "nav_menu": {
    "height": "279px",
diff --git a/images/end_to_end_project/income_category_hist.png b/images/end_to_end_project/income_category_hist.png
new file mode 100644
index 0000000..e41ab93
Binary files /dev/null and b/images/end_to_end_project/income_category_hist.png differ