Add primes computation code. Add tests