python code to check if a number is prime