diff --git a/examples/plot_kymograph.py b/examples/plot_kymograph.py
index c6d468be22203563b4122401e72bca5f14caa6d3..0a314065a14734b3c618989691c84f0b940d9d90 100644
--- a/examples/plot_kymograph.py
+++ b/examples/plot_kymograph.py
@@ -1,30 +1,27 @@
-from wela.dataloader import dataloader
-from wela.plotting import kymograph, plot_lineage
-
-# from genutils import figs2pdf
 import matplotlib.cm
-
+from wela.figs2pdf import figs2pdf
+from wela.dataloader import dataloader, sort_df
+from wela.plotting import kymograph, plot_lineage
 
 datasets = ["Pdr5_3_11_22", "Pdr5_flc_10ugml_1676"]
 for dataname in datasets:
-    dl = dataloader(".", "/Users/pswain/wip/tsv_data")
+    dl = dataloader(".", "/Users/pswain/wip/tsv_data/pdr_with_gr/")
     dl.load(dataname, use_tsv=True)
-    df = dl.get_sorted_df("median_GFP")
+    sdf = sort_df(dl.df, "median_GFP")
 
     title = dataname.split(".")[0]
-    kymograph(df, hue="buddings", title=title)
-    kymograph(df, hue="median_GFP", title=title)
+    kymograph(sdf, hue="buddings", title=title)
+    kymograph(sdf, hue="median_GFP", title=title)
     kymograph(
-        df,
+        sdf,
         hue="bud_growth_rate",
         title=title,
         vmin=-50,
         vmax=50,
         cmap=matplotlib.cm.Spectral,
     )
-    # df.id = df.id.astype("object")
-    # df.to_csv(dataname + ".tsv", sep="\t", index=False)
-# figs2pdf("fluconazole")
+
 
 if False:
+    figs2pdf("fluconazole")
     plot_lineage(dl.ids[34], dl.df)