Simple Code For Prime Number In Python