Wrote test for primes