Add function 'is_prime()'