結果 : check if a number is prime in python code