diff --git a/requirements.txt b/requirements.txt
index 3c4d3c0..0806932 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 numpy
 scipy
 scikit-learn
+matplotlib
 networkx
 pgmpy
diff --git a/setup.py b/setup.py
index 3180852..0a0e2ea 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,14 @@ URL = "https://github.com/doctorado-ml/bayesclass"
 LICENSE = "MIT"
 DOWNLOAD_URL = "https://github.com/doctorado-ml/bayesclass"
 VERSION = __version__
-INSTALL_REQUIRES = ["numpy", "scipy", "scikit-learn", "pgmpy", "networkx"]
+INSTALL_REQUIRES = [
+    "numpy",
+    "scipy",
+    "scikit-learn",
+    "pgmpy",
+    "networkx",
+    "matplotlib",
+]
 CLASSIFIERS = [
     "Intended Audience :: Science/Research",
     "Intended Audience :: Developers",