add isPrime function