diff --git a/ultranest/pathsampler.py b/ultranest/pathsampler.py
index 85b3c18..b0aaf55 100644
--- a/ultranest/pathsampler.py
+++ b/ultranest/pathsampler.py
@@ -1,4 +1,4 @@
-"""MCMC-like step sampling on a trajectory
+"""MCMC-like step sampling on a trajectory.
 
 These features are experimental.
 """
@@ -243,7 +243,7 @@ def set_gradient(self, grad_function):
         print("set gradient function to %s" % grad_function.__name__)
 
         def plot_gradient_wrapper(x, plot=False):
-            """wrapper that makes plots (when desired)"""
+            """Make plot while computing gradient (optionally)."""
             v = grad_function(x)
             if plot:
                 plt.plot(x[0], x[1], '+ ', color='k', ms=10)