Prime Number Code Using Python