Import astropy units

tofil.py

@@ -5,6 +5,7 @@ import numpy as np
 import filterbank
 import sigproc
 from astropy.time import Time
+from astropy import units as u
 
 import os.path
 import os