diff --git a/Makefile b/Makefile
index cc0bb675b26cafb0f7bb63197affde6fbca23b98..a216e69f6aa725a27bf44ff89bd8cf876759b561 100644
--- a/Makefile
+++ b/Makefile
@@ -18,8 +18,11 @@ clean:
 
 test: all
 	@ time -f "%C : %E seconds" ./a.out $(ROUNDS) > /dev/null
-	@ [ -n "$$(which python3)" ] && time -f "%C : %E seconds" ./prime.py $(ROUNDS) > /dev/null
+	@ [ -n "$$(which python)" ] && time -f "%C : %E seconds" ./prime.py $(ROUNDS) > /dev/null
+	@ [ -n "$$(which python3)" ] && time -f "%C : %E seconds" python3 prime.py $(ROUNDS) > /dev/null
+	@ [ -n "$$(which pypy)" ] && time -f "%C : %E seconds" pypy  prime.py $(ROUNDS) > /dev/null
 	@ [ -n "$$(which lua)" ] && time -f "%C : %E seconds" ./prime.lua $(ROUNDS) > /dev/null
+	@ [ -n "$$(which luajit)" ] && time -f "%C : %E seconds" luajit prime.lua $(ROUNDS) > /dev/null
 	@ [ -n "$$(which php)" ] && time -f "%C : %E seconds" ./prime.php $(ROUNDS) > /dev/null
 	@ [ -n "$$(which php7.0)" ] && time -f "%C : %E seconds" php7.0 prime.php $(ROUNDS) > /dev/null
 	@ [ -n "$$(which ruby)" ] && time -f "%C : %E seconds" ./prime.rb $(ROUNDS) > /dev/null
diff --git a/prime.py b/prime.py
index 004ffa2300fe48e26f0519a085d44e9780f4ec6e..78fbf9b933bc5a5e44ef2665af5f5486e5667076 100755
--- a/prime.py
+++ b/prime.py
@@ -1,9 +1,13 @@
-#!/usr/bin/env python3
+#!/usr/bin/env python
+from __future__ import (absolute_import, division,
+                        print_function, unicode_literals)
+
+
 def is_prime(x):
     from math import sqrt, ceil
     if sqrt(x).is_integer():
         return False
-    for i in range(2, ceil(sqrt(x))):
+    for i in range(2, int(ceil(sqrt(x)))):
         if x % i == 0:
             return False
     return True