diff --git a/code/predict.py b/code/predict.py
index fc57d8f..cf26ef0 100644
--- a/code/predict.py
+++ b/code/predict.py
@@ -169,7 +169,7 @@ def predict(files, model_path, output_dir, format, use_lstm=True):
             #continue
         #'''
 
-        sys.stdout.write('%s\n' % '-' * 30)
+        sys.stdout.write('%s\n' % ('-' * 30))
         sys.stdout.write('\n\t%d of %d\n' % (i+1,n))
         sys.stdout.write('\t%s\n\n' % txt)
 
diff --git a/code/tools.py b/code/tools.py
index a6597d4..8a3a094 100644
--- a/code/tools.py
+++ b/code/tools.py
@@ -10,6 +10,7 @@
 import os
 import errno
 import string
+import math
 import re
 import pickle
 import numpy as np