Prime Number Checking Python Code