Improve algorithm

Author: Charliiee

Math/Prime/python/prime.py

@@ -3,9 +3,14 @@
 import math
 
 def is_prime(number):
-    if (number <= 1):
+    if number <= 1:
         return False
-    for i in range(2, 1 + int(math.sqrt(number))):
+    if number == 2:
+        return True
+    if number % 2 == 0:
+        return False
+
+    for i in range(3, 1 + int(math.sqrt(number)), 2):
         if (number % i) == 0:
             return False
     return True