diff --git a/README.md b/README.md
index a238b567..2eb08aef 100644
--- a/README.md
+++ b/README.md
@@ -24,12 +24,12 @@ machine and on the web.
 
 ## Features 🧩
 
-- Modern, **declarative UI** framework
-- **100% Python** - Zero HTML, CSS, or JavaScript required
-- Over **50 Built-in components** for common UI elements, such as `rio.Switch`, `rio.Button`, and `rio.Text`, and many more
-- Integrates with **modern Python tooling**: Thanks to being **entirely Type Safe** editors can give you instant suggestions and highlight problems right away
-- Apps can run **both locally and on the web**
-- **Open Source & Free forever**
+-   Modern, **declarative UI** framework
+-   **100% Python** - Zero HTML, CSS, or JavaScript required
+-   Over **50 Built-in components** for common UI elements, such as `rio.Switch`, `rio.Button`, and `rio.Text`, and many more
+-   Integrates with **modern Python tooling**: Thanks to being **entirely Type Safe** editors can give you instant suggestions and highlight problems right away
+-   Apps can run **both locally and on the web**
+-   **Open Source & Free forever**
 
 ## Installation 🛠️
 
@@ -70,7 +70,6 @@ client = AsyncOpenAI()
 
 
 class DallEPage(rio.Component):
-
     prompt: str = ""
     image_url: str = ""
     is_loading: bool = False
@@ -146,6 +145,7 @@ app = rio.App(
         ),
     ],
 )
+```
 
 ## Status: In Development 🚧
 
@@ -154,6 +154,4 @@ experience. If you have any feedback, please let us know on our Discord server.
 
 ## Contributing 🤝
 
-Write how someone can contribute to the project. :)
-
-```
+TODO: Write how someone can contribute to the project. :)
diff --git a/scripts/check_docs.py b/scripts/check_docs.py
index 59da5e58..d5acbcb9 100644
--- a/scripts/check_docs.py
+++ b/scripts/check_docs.py
@@ -12,17 +12,16 @@ from typing import *  # type: ignore
 # annotations can fail if they're not installed. Import them explicitly here to produce more obvious error messages
 from revel import *  # type: ignore
 
-# Make sure `rio_docs` can be imported
+# Make sure `rio.docs` can be imported
 sys.path.append(str(Path(__file__).absolute().parent.parent))
 
-import rio_docs
-import website
+import rio_website.article_models
 
-import rio
+import rio.docs
 
 
 def check_function(
-    docs: rio_docs.FunctionDocs,
+    docs: rio.docs.FunctionDocs,
     owning_cls: type | None,
 ) -> None:
     qualname = (
@@ -62,7 +61,7 @@ def check_function(
             )
 
 
-def check_class(cls: type, docs: rio_docs.ClassDocs) -> None:
+def check_class(cls: type, docs: rio.docs.ClassDocs) -> None:
     # Run checks
     if docs.summary is None:
         warning(f"Docstring for `{docs.name}` is missing a short description")
@@ -95,7 +94,7 @@ def main() -> None:
     # Find all items that should be documented
     print_chapter("Looking for items needing documentation")
     target_items: list[type | Callable[..., Any]] = list(
-        rio_docs.custom.find_items_needing_documentation()
+        rio.docs.custom.find_items_needing_documentation()
     )
 
     print(f"Found {len(target_items)} items")
@@ -106,30 +105,30 @@ def main() -> None:
         # Classes / Components
         if inspect.isclass(item):
             # Fetch the docs
-            docs = rio_docs.ClassDocs.parse(item)
+            docs = rio.docs.ClassDocs.parse(item)
 
             # Post-process them as needed
             if isinstance(item, rio.Component):
-                rio_docs.custom.postprocess_component_docs(docs)
+                rio.docs.custom.postprocess_component_docs(docs)
             else:
-                rio_docs.custom.postprocess_class_docs(docs)
+                rio.docs.custom.postprocess_class_docs(docs)
 
             check_class(item, docs)
 
         else:
             assert inspect.isfunction(item), item
 
-            docs = rio_docs.FunctionDocs.parse(item)
+            docs = rio.docs.FunctionDocs.parse(item)
 
             check_function(docs, None)
 
     # Make sure all items are displayed Rio's documentation
     visited_item_names: set[str] = set()
 
-    for entry in website.structure.DOCUMENTATION_STRUCTURE_LINEAR:
+    for entry in rio_website.structure.DOCUMENTATION_STRUCTURE_LINEAR:
         section_name, section_url, builder = entry
 
-        if not isinstance(builder, website.article_models.ArticleBuilder):
+        if not isinstance(builder, rio_website.article_models.ArticleBuilder):
             continue
 
         visited_item_names.add(builder.component_class.__name__)