diff --git a/__init__.py b/__init__.py
index 390929e..218bb61 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,3 @@
-
 """ Make simple, pretty Sankey Diagrams """
 
 # allow repo itself to be loaded as a package
diff --git a/ausankey/__init__.py b/ausankey/__init__.py
index 78a6a95..92384f1 100644
--- a/ausankey/__init__.py
+++ b/ausankey/__init__.py
@@ -1,4 +1,4 @@
-
 """ Make simple, pretty Sankey Diagrams """
 
+from .ausankey import sankey
 
diff --git a/pyproject.toml b/pyproject.toml
index a5d4064..5979f53 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -87,3 +87,6 @@ test = "python -m pytest"
 test-cov = "coverage run -m pytest"
 coverall = "coveralls --service=github"
 cov = ["test-cov", "coverall"]
+
+[tool.ruff]
+exclude = ["__init__.py"]
diff --git a/sankey_example_minimal.py b/sankey_example_minimal.py
index 9f1f875..318e42b 100644
--- a/sankey_example_minimal.py
+++ b/sankey_example_minimal.py
@@ -1,6 +1,6 @@
-import pandas as pd
-import matplotlib.pyplot as plt
 import ausankey as sky
+import matplotlib.pyplot as plt
+import pandas as pd
 
 data = pd.DataFrame([
   ("a",1.0,"ab",2.0,"a",1.0),
diff --git a/tests/generic_test.py b/tests/generic_test.py
index cfa5d03..7a5de56 100644
--- a/tests/generic_test.py
+++ b/tests/generic_test.py
@@ -1,4 +1,3 @@
-
 import os
 import unittest