Simple Python Code For Prime Number