diff --git a/.gitignore b/.gitignore
index 976ba81..2155dda 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
 *.pyc
-commenting_code_model/visualizer/random_forest_model_tree_*
+commenting_code_model/visualizer/random_forest_model_*
+commenting_code_model/random_forest_model_*
+__pycache__
diff --git a/commenting_code_model/visualizer/requirements.txt b/commenting_code_model/visualizer/requirements.txt
deleted file mode 100644
index abecf5d..0000000
--- a/commenting_code_model/visualizer/requirements.txt
+++ /dev/null
@@ -1 +0,0 @@
-graphviz
\ No newline at end of file