Python Prime Number Example