写一个判断素数的函数，函数名为is_prime，在主函数输入一个整数，输出是否是素数的消息。

一个数

如果是素数输出prime 如果不是输出not prime