Add docstrings to primes, closes #2.