diff --git a/clrs/__init__.py b/clrs/__init__.py
index 5cec1f69..c783fede 100644
--- a/clrs/__init__.py
+++ b/clrs/__init__.py
@@ -34,7 +34,7 @@
 from clrs._src.specs import Stage
 from clrs._src.specs import Type
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 __all__ = (
     "build_sampler",
diff --git a/setup.py b/setup.py
index 9bf3beb4..ed242e76 100644
--- a/setup.py
+++ b/setup.py
@@ -48,11 +48,11 @@ def _parse_requirements(path):
     url='https://github.com/deepmind/clrs',
     license='Apache 2.0',
     author='DeepMind',
-    description=('A library of reinforcement learning building blocks in JAX.'),
+    description=('The CLRS Algorithmic Reasoning Benchmark.'),
     long_description=open(os.path.join(_CURRENT_DIR, 'README.md')).read(),
     long_description_content_type='text/markdown',
     author_email='clrs-dev@google.com',
-    keywords='reinforcement-learning python machine learning',
+    keywords='python machine learning',
     packages=find_namespace_packages(exclude=['*_test.py']),
     install_requires=_parse_requirements(
         os.path.join(_CURRENT_DIR, 'requirements', 'requirements.txt')),