diff --git a/bind/run_pbwo4.py b/bind/run_pbwo4.py
index 5eb80e4..ce7218e 100644
--- a/bind/run_pbwo4.py
+++ b/bind/run_pbwo4.py
@@ -7,8 +7,11 @@ output_dir = os.path.dirname(os.path.abspath(__file__))
 gd = GeometryDescriptor()
 gd.addLayer(20.0, "G4_PbWO4", True, 10, 10)
 
-df = run_batch(gd, 10, "e-", 1.0)
+hits, steps = run_batch(gd, 10, "e-", 1.0, return_steps=True)
 
-out = os.path.join(output_dir, "pbwo4_10events.pkl")
-df.to_pickle(out)
-print(f"Saved {len(df)} events to {out}")
+hits_out  = os.path.join(output_dir, "pbwo4_10events_hits.pkl")
+steps_out = os.path.join(output_dir, "pbwo4_10events_steps.pkl")
+hits.to_pickle(hits_out)
+steps.to_pickle(steps_out)
+print(f"Saved {len(hits)} events to {hits_out}")
+print(f"Saved {len(steps)} steps  to {steps_out}")