diff --git a/python/examples/repetition_code.py b/python/examples/repetition_code.py
index 5c64be1..945e826 100644
--- a/python/examples/repetition_code.py
+++ b/python/examples/repetition_code.py
@@ -1,6 +1,7 @@
 import argparse
 import numpy as np
 import pandas as pd
+import matplotlib.pyplot as plt
 
 # autopep8: off
 import sys
@@ -104,25 +105,6 @@ def track_path(args):
 
     plt.show()
 
-    # df = pd.DataFrame({"x1b": ys_start[:, 0],
-    #                    "x2b": ys_start[:, 1],
-    #                    "tb":  ys_start[:, 2],
-    #                    "x1p": ys_prime[:, 0],
-    #                    "x2p": ys_prime[:, 1],
-    #                    "tp":  ys_prime[:, 2],
-    #                    "x1e": ys_hat_e[:, 0],
-    #                    "x2e": ys_hat_e[:, 1],
-    #                    "te":  ys_hat_e[:, 2],
-    #                    "x1n": ys[:, 0],
-    #                    "x2n": ys[:, 1],
-    #                    "tn":  ys[:, 2]
-    #                    })
-    #
-    # if args.output:
-    #     df.to_csv(args.output, index=False)
-    # else:
-    #     print(df)
-
 
 def main():
     parser = argparse.ArgumentParser(
@@ -139,7 +121,6 @@ def main():
                         default=0.01, help="Convergence threshold for Newton corrector")
     parser.add_argument("-s", "--sigma", type=int, default=1,
                         help="Direction in which the path is traced")
-    parser.add_argument("-o", "--output", type=str, help="Output csv file")
     parser.add_argument("-n", "--num-iterations", type=int, default=20,
                         help="Number of iterations of the example program to run")