結果 : define a prime number in python