diff --git a/pysuperconductor/os_gap_calc.py b/pysuperconductor/os_gap_calc.py
index bf886ec..944ef96 100644
--- a/pysuperconductor/os_gap_calc.py
+++ b/pysuperconductor/os_gap_calc.py
@@ -32,11 +32,16 @@ def gap_integral(
 	return integrate.quad(integrand, 0, debye_frequency)[0]
 
 
-def equilibrium_gap(temp, debye_frequency, nv):
+def equilibrium_gap(temp: float, debye_frequency: float, nv: float) -> float:
 	return find_gap(temp, 0, debye_frequency, nv)
 
 
-def find_gap(temp, mu_star, debye_frequency, nv):
+def find_gap(
+	temp: float,
+	mu_star: float,
+	debye_frequency: float,
+	nv: float
+) -> float:
 	nv_inv = 1 / nv
 	sol = scipy.optimize.root(
 		lambda d: gap_integral(temp, d, mu_star, debye_frequency) - nv_inv,