1 files changed, 10 insertions, 10 deletions
diff --git a/examples/example_diffusion.py b/examples/example_diffusion.py
index dc2d05e..8826d16 100644
--- a/examples/example_diffusion.py
+++ b/examples/example_diffusion.py
@@ -11,23 +11,23 @@ import numpy as np
 import matplotlib.pyplot as plt
 
 # Import transivent functions
-from transivent import (
-    process_file,
-    get_final_events,
+from transivent import configure_logging, detect
+
+# Import building blocks from submodules
+from transivent.analysis import (
+    detect_events,
+    calculate_initial_background,
+    estimate_noise,
+    analyze_thresholds,
+)
+from transivent.event_processor import (
     process_events_for_diffusion,
     extract_event_waveforms,
     calculate_msd_parallel,
     calculate_acf,
     fit_diffusion_linear,
     plot_diffusion_comparison,
-    detect_events,
-    calculate_initial_background,
-    estimate_noise,
-    analyze_thresholds,
 )
-
-# Configure logging
-from transivent import configure_logging
 configure_logging("INFO")