diff --git a/src/cellmap_models/__init__.py b/src/cellmap_models/__init__.py
index 1be6c6a..cd9c858 100644
--- a/src/cellmap_models/__init__.py
+++ b/src/cellmap_models/__init__.py
@@ -3,4 +3,4 @@
 """
 
 from .utils import download_url_to_file
-from .pytorch import cosem
\ No newline at end of file
+from .pytorch import cosem
diff --git a/src/cellmap_models/pytorch/__init__.py b/src/cellmap_models/pytorch/__init__.py
index 5ef22e7..d95c684 100755
--- a/src/cellmap_models/pytorch/__init__.py
+++ b/src/cellmap_models/pytorch/__init__.py
@@ -1 +1 @@
-from . import cosem
\ No newline at end of file
+from . import cosem
diff --git a/src/cellmap_models/pytorch/cosem/__init__.py b/src/cellmap_models/pytorch/cosem/__init__.py
index dd7cfb1..a07c6c8 100755
--- a/src/cellmap_models/pytorch/cosem/__init__.py
+++ b/src/cellmap_models/pytorch/cosem/__init__.py
@@ -1,4 +1,5 @@
 from .load_model import load_model
+
 models_list = [
     "setup04/1820500.pth",
     "setup04/625000.pth",
@@ -11,4 +12,4 @@
     "setup36/500000.pth",
     "setup45/1634500.pth",
     "setup45/625000.pth",
-]
\ No newline at end of file
+]
diff --git a/src/cellmap_models/pytorch/cosem/setup36/model.py b/src/cellmap_models/pytorch/cosem/setup36/model.py
index c56daf6..78eae57 100755
--- a/src/cellmap_models/pytorch/cosem/setup36/model.py
+++ b/src/cellmap_models/pytorch/cosem/setup36/model.py
@@ -36,4 +36,3 @@
     if not (Path(__file__).parent / f"{name}").exists():
         print(f"Downloading {name} from {url}")
         download_url_to_file(url, str(Path(__file__).parent / f"{name}.pth"))
-
diff --git a/src/cellmap_models/pytorch/cosem/setup45/model.py b/src/cellmap_models/pytorch/cosem/setup45/model.py
index 638de9e..1fa8cb7 100755
--- a/src/cellmap_models/pytorch/cosem/setup45/model.py
+++ b/src/cellmap_models/pytorch/cosem/setup45/model.py
@@ -36,4 +36,3 @@
     if not (Path(__file__).parent / f"{name}").exists():
         print(f"Downloading {name} from {url}")
         download_url_to_file(url, str(Path(__file__).parent / f"{name}.pth"))
-