Fix isPrime function