how to check if a number is prime in python
import math
def prime(x):
    if x == 2 or x == 3 or x==5:
        return True
    if x == 1 or x == 4:
        return False
    num = 2
    a= math.sqrt(x)+4 
    while a > num:
        if x % num == 0:
            return False
        else:
            num = num + 1
    return True