From e7919d309a9c8966d11a095cce6e70632ffb4b40 Mon Sep 17 00:00:00 2001
From: erdogant <erdogant@gmail.com>
Date: Sat, 12 Mar 2022 20:31:22 +0100
Subject: [PATCH] 0.1.6

---
 kaplanmeier/__init__.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/kaplanmeier/__init__.py b/kaplanmeier/__init__.py
index b051a76..ac7c330 100644
--- a/kaplanmeier/__init__.py
+++ b/kaplanmeier/__init__.py
@@ -6,7 +6,7 @@
 
 __author__ = 'Erdogan Tasksen'
 __email__ = 'erdogant@gmail.com'
-__version__ = '0.1.5'
+__version__ = '0.1.6'
 
 # module level doc-string
 __doc__ = """
@@ -18,13 +18,23 @@
 
 Example
 ----------
+>>> # Import library
 >>> import kaplanmeier as km
+>>>
+>>> # Example data
 >>> df = km.example_data()
->>> out = km.fit(df['time'], df['Died'], df['group'])
->>> km.plot(out)
-
+>>>
+>>> # Fit
+>>> results = km.fit(df['time'], df['Died'], df['group'])
+>>>
+>>> # Plot
+>>> km.plot(results)
+>>>
 >>> km.plot(out, cmap='Set1', cii_lines=True, cii_alpha=0.05)
 >>> km.plot(out, cmap=[(1, 0, 0),(0, 0, 1)])
 >>> km.plot(out, cmap='Set1', methodtype='custom')
+>>>
+>>> results['logrank_P']
+>>> results['logrank_Z']
 
 """