program to check prime number in python