Prime Number Python Simple Code