Prime Number Checker Python Code