Simple Prime Number Code In Python