nth_prime function