Funzione per trovare numeri primi

def prime(n):
 if n <= 3:
   return "primo"
 m = 2
 while m <= (n*m):
   if n % m == 0:
     return "non primo"
   m = m+1
   return "primo"
 
numero = input('Inserisci il numero: ')
print prime(numero)