diff --git a/causal_curve/core.py b/causal_curve/core.py
index bd492b1..dd3c64d 100644
--- a/causal_curve/core.py
+++ b/causal_curve/core.py
@@ -12,7 +12,7 @@ class Core:
     def __init__(self):
         pass
 
-    __version__ = "1.0.3"
+    __version__ = "1.0.4"
 
     def get_params(self):
         """Returns a dict of all of the object's user-facing parameters
diff --git a/docs/TMLE_Regressor.rst b/docs/TMLE_Regressor.rst
index 0243301..9c2307a 100644
--- a/docs/TMLE_Regressor.rst
+++ b/docs/TMLE_Regressor.rst
@@ -62,11 +62,12 @@ All we do now is employ the TMLE_Regressor class, with mostly default settings:
 
 >>> from causal_curve import TMLE_Regressor
 tmle = TMLE_Regressor(
-    random_seed=111
+    random_seed=111,
+		bandwidth=10
 )
 
->>> tmle.fit(T = df['Treatment'], X = df[['X_1', 'X_2']], y = df['Outcome'])
-gps_results = tmle.calculate_CDRC(0.99)
+>>> tmle.fit(T = df['treatment'], X = df[['x1', 'x2']], y = df['outcome'])
+gps_results = tmle.calculate_CDRC(0.95)
 
 The resulting dataframe contains all of the data you need to generate the following plot:
 
diff --git a/docs/changelog.rst b/docs/changelog.rst
index d3bb0c0..264e2d3 100644
--- a/docs/changelog.rst
+++ b/docs/changelog.rst
@@ -4,6 +4,9 @@
 Change Log
 ==========
 
+Version 1.0.4
+-------------
+- Fixed TMLE plot and code errors in documentation
 
 Version 1.0.3
 -------------
diff --git a/docs/conf.py b/docs/conf.py
index d1fc56b..d9d940e 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -23,7 +23,7 @@
 author = "Roni Kobrosly"
 
 # The full version, including alpha/beta/rc tags
-release = "1.0.3"
+release = "1.0.4"
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/imgs/tmle_plot.png b/imgs/tmle_plot.png
index e5397fe..eb85c01 100644
Binary files a/imgs/tmle_plot.png and b/imgs/tmle_plot.png differ
diff --git a/setup.py b/setup.py
index 15a26f6..3e7196f 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="causal-curve",
-    version="1.0.3",
+    version="1.0.4",
     author="Roni Kobrosly",
     author_email="roni.kobrosly@gmail.com",
     description="A python library with tools to perform causal inference using \