diff --git a/py4DSTEM/__init__.py b/py4DSTEM/__init__.py
index c721dcea2..59b2cd8d4 100644
--- a/py4DSTEM/__init__.py
+++ b/py4DSTEM/__init__.py
@@ -52,7 +52,8 @@
     BraggVectorMap,
 )
 
-from py4DSTEM.process import classification 
+from py4DSTEM.process import classification
+
 # import (
 #     BraggVectorClassification,
 #     ConvergenceWarning,
@@ -76,7 +77,8 @@
 
 
 # ptycho
-from py4DSTEM.process import phase 
+from py4DSTEM.process import phase
+
 # import (
 #     DPCReconstruction,
 #     MixedstatePtychographicReconstruction,
@@ -102,6 +104,7 @@
 
 # TODO WPF
 from py4DSTEM.process import wholepatternfit
+
 #  import (
 #     ComplexOverlapKernelDiskLattice,
 #     DCBackground,