prime number code in python