diff --git a/ismfeed/__init__.py b/ismfeed/__init__.py
index e69de29..9fe3e6c 100644
--- a/ismfeed/__init__.py
+++ b/ismfeed/__init__.py
@@ -0,0 +1,5 @@
+from .ismfeed import (
+    get_sinks,
+    compute_sfr,
+    plot_mass_sfr,
+)
diff --git a/ismfeed/ismfeed.py b/ismfeed/ismfeed.py
index bd34478..3c7a1b4 100644
--- a/ismfeed/ismfeed.py
+++ b/ismfeed/ismfeed.py
@@ -9,7 +9,7 @@ import matplotlib as mpl
 
 import matplotlib.ticker as tick
 import matplotlib.pyplot as plt
-from matplotlib import FuncFormatter
+from matplotlib.ticker import FuncFormatter
 
 from utils.units import convert_exp