prime generator draft in python