Prime Number Condition For Python