Prime Condition In Python