Program to check if a given number is a Prime Number or not

def is_prime(n):
    if n < 2 or (n % 2 == 0 and n > 2): return False
    return all(n % i for i in range(3, int(n**0.5) + 1, 2))

num = int(input("Enter a number: "))
print(f"{num} is {'a Prime' if is_prime(num) else 'NOT a Prime'} number.")