prime or not (2nd meth)(f)

def prime(m):
    for i in range(2,m):
        if m%i==0:
            print(m,'is not a prime')
            break
    else:
        print(m,' is a prime')
m=int(input('enter a number:'))
prime(m)

#output
enter a number:5
5 is a prime

enter a number:4
4 is not a prime