1 files changed, 2 insertions, 1 deletions
diff --git a/tools.py b/tools.py
index 2a44104..6835228 100644
--- a/tools.py
+++ b/tools.py
@@ -5,9 +5,10 @@ import datetime
 def timeit(func):
     def f(*args, **kwargs):
         t0 = time.time()
-        func(*args, **kwargs)
+        retval = func(*args, **kwargs)
         elapsed = time.time() - t0
         print '%s elapsed in %s().' % (datetime.timedelta(seconds=elapsed), func.__name__)
+        return retval
     return f
 
 def read_csv(filename):