diff --git a/code-server/README.md b/code-server/README.md
index 9b63912..6f068ed 100644
--- a/code-server/README.md
+++ b/code-server/README.md
@@ -11,15 +11,14 @@ tags: [helper, ide]
 
 Automatically install [code-server](https://github.com/coder/code-server) in a workspace, create an app to access it via the dashboard, install extensions, and pre-configure editor settings.
 
-![Screenshot 1](https://github.com/coder/code-server/raw/main/docs/assets/screenshot-1.png?raw=true)
-![Screenshot 2](https://github.com/coder/code-server/raw/main/docs/assets/screenshot-2.png?raw=true)
-
-## Highlights
+```hcl
+module "code-server" {
+    source = "https://registry.coder.com/modules/code-server"
+    agent_id = coder_agent.example.id
+}
+```
 
-- Code on any device with a consistent development environment
-- Use cloud servers to speed up tests, compilations, downloads, and more
-- Preserve battery life when you're on the go; all intensive tasks run on your
-  server
+![Screenshot 1](https://github.com/coder/code-server/raw/main/docs/assets/screenshot-1.png?raw=true)
 
 ## Examples
 
@@ -30,6 +29,7 @@ Install the Dracula theme from [OpenVSX](https://open-vsx.org/):
 ```hcl
 module "code-server" {
     source = "https://registry.coder.com/modules/code-server"
+    agent_id = coder_agent.example.id
     extensions = [
         "dracula-theme.theme-dracula"
     ]
@@ -45,6 +45,7 @@ Configure VS Code's [settings.json](https://code.visualstudio.com/docs/getstarte
 ```hcl
 module "settings" {
     source = "https://registry.coder.com/modules/code-server"
+    agent_id = coder_agent.example.id
     extensions = [ "dracula-theme.theme-dracula" ]
     settings = {
         "workbench.colorTheme" = "Dracula"
@@ -59,6 +60,7 @@ Just run code-server in the background, don't fetch it from GitHub:
 ```hcl
 module "settings" {
     source = "https://registry.coder.com/modules/code-server"
+    agent_id = coder_agent.example.id
     offline = true
 }
 ```
\ No newline at end of file