Prime num

prime = True
x = int(input("Enter a num:"))
def funct(a):
    for i in range(2,a):
        if a%i == 0:
            return print("Not a prime number")
        else:
            return print("prime number")
funct(x)