Add tests for primality