diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py
index 8c9ab07..d0434e6 100644
--- a/main/flaredetector/flaredetector.py
+++ b/main/flaredetector/flaredetector.py
@@ -1,4 +1,7 @@
 import numpy as np
+from scipy.optimize import curve_fit
+from .util import *
+
 
 def GaussExpo(x, Peak, l, w):
     sigma = 1/w