Add updated function for happy_primes