diff --git a/src/leibnetz/nets/__init__.py b/src/leibnetz/nets/__init__.py
index 4c0dd6d..4c42c5b 100644
--- a/src/leibnetz/nets/__init__.py
+++ b/src/leibnetz/nets/__init__.py
@@ -1,5 +1,12 @@
 from .attentive_scalenet import build_attentive_scale_net
 from .scalenet import build_scale_net
 from .unet import build_unet
+from .bio import (
+    convert_to_bio,
+    convert_to_backprop,
+    HebbsRule,
+    KrotovsRule,
+    OjasRule,
+)
 
 # from .resnet import build_resnet
diff --git a/src/leibnetz/nets/bio.py b/src/leibnetz/nets/bio.py
index fa6b555..21b39bf 100644
--- a/src/leibnetz/nets/bio.py
+++ b/src/leibnetz/nets/bio.py
@@ -245,11 +245,3 @@ def convert_to_backprop(model: LeibNet):
 
 
 # %%
-from leibnetz.nets import build_unet as leibnetz_unet
-
-model = convert_to_bio(leibnetz_unet(), KrotovsRule(), learning_rate=0.1)
-inputs = model.get_example_inputs(device="cuda")
-model = model.to("cuda")
-# %%
-outputs = model(inputs)
-# %%