prime number program in python