[python] Checking if a number is a prime number in Python