Python program for prime number