def prime(a):
    if a < 2:
        return False
if a % 2 == 0:
    return False
if a == 3 or a == 5 or a == 7:
    return True
for n in range(3, int(a ** 0.5) + 1):
    if a % n == 0:
        pass
Difficulty with python prime number check