結果 : create a function to check prime number in python