diff --git a/README.md b/README.md
index 8cc2e60..2ee57c8 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,11 @@
+# Welcome to AiiDA WorkGraph Web UI!
+[![PyPI version](https://badge.fury.io/py/aiida-workgraph-web-ui.svg)](https://badge.fury.io/py/aiida-workgraph-web-ui)
+[![Docs status](https://readthedocs.org/projects/aiida-workgraph-web-ui/badge)](http://aiida-workgraph-web-ui.readthedocs.io/)
+[![Unit test](https://github.com/aiidateam/aiida-workgraph-web-ui/actions/workflows/ci.yml/badge.svg)](https://github.com/aiidateam/aiida-workgraph-web-ui/actions/workflows/ci.yml)
 
 
+Web UI to visualize and manage the AiiDA WorkGraph.
+
 ## Frontend
 Use `rete-kit` to create the frontend, using `react`
 
diff --git a/aiida_workgraph_web_ui/__init__.py b/aiida_workgraph_web_ui/__init__.py
index 36e61c6..3dc1f76 100644
--- a/aiida_workgraph_web_ui/__init__.py
+++ b/aiida_workgraph_web_ui/__init__.py
@@ -1,2 +1 @@
-
 __version__ = "0.1.0"
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 6f4b0cd..a450687 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -12,8 +12,6 @@
 #
 import os
 import sys
-import shutil
-from pathlib import Path
 
 sys.path.insert(0, os.path.abspath("../.."))
 
diff --git a/pyproject.toml b/pyproject.toml
index a7af23b..b8328e7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,8 +27,8 @@ requires-python = ">=3.9"
 dependencies = [
     "numpy~=1.21",
     "scipy",
-    "ase",
     "aiida-core>=2.3",
+    "aiida_workgraph",
     "cloudpickle",
     "fastapi",
     "uvicorn",