結果 : write a function to check if a number is prime python