diff --git a/.github/workflows/ci_cd.yml b/.github/workflows/ci_cd.yml
index c240cd8..21865a0 100644
--- a/.github/workflows/ci_cd.yml
+++ b/.github/workflows/ci_cd.yml
@@ -1,12 +1,32 @@
-name: CI/CD Workflow
+name: CI/CD
 
 on:
   push:
-    branches:
-      - main
 
 jobs:
+  ci:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v3
+
+      - uses: actions/setup-node@v3
+        with:
+          node-version: 20
+
+      - name: Install dependencies
+        run: yarn install
+
+      - name: Lint
+        run: yarn lint
+
+      - name: Test
+        run: yarn test
+
   build-and-push:
+    needs: ci
+    if: github.ref == 'refs/heads/main'
     runs-on: ubuntu-latest
 
     steps:
@@ -35,6 +55,14 @@ jobs:
           push: true
           tags: docker.io/${{ secrets.DOCKERHUB_USERNAME }}/linux-odyssey:app
 
+      - name: Build base image
+        uses: docker/build-push-action@v2
+        with:
+          context: .
+          file: ./quests/Dockerfile
+          push: false
+          tags: docker.io/${{ secrets.DOCKERHUB_USERNAME }}/linux-odyssey:base
+
       - name: Build quest helloworld
         uses: docker/build-push-action@v2
         with:
diff --git a/.github/workflows/eslint.yml b/.github/workflows/eslint.yml
deleted file mode 100644
index 01f475e..0000000
--- a/.github/workflows/eslint.yml
+++ /dev/null
@@ -1,22 +0,0 @@
-name: ESLint Checks
-
-on:
-  push:
-
-jobs:
-  lint:
-    runs-on: ubuntu-latest
-
-    steps:
-      - name: Checkout
-        uses: actions/checkout@v3
-
-      - uses: actions/setup-node@v3
-        with:
-          node-version: 20
-
-      - name: Install dependencies
-        run: yarn install
-
-      - name: Lint
-        run: yarn lint
diff --git a/.gitignore b/.gitignore
index a389d25..8534ca9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -9,6 +9,7 @@ lerna-debug.log*
 .bash_history
 .viminfo
 *.swp
+home
 
 # Diagnostic reports (https://nodejs.org/api/report.html)
 report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json
diff --git a/app/Dockerfile b/app/Dockerfile
index dd77845..30a17f7 100644
--- a/app/Dockerfile
+++ b/app/Dockerfile
@@ -1,14 +1,18 @@
 # Build stage
 FROM node as build-stage
 WORKDIR /app
-COPY app/package.json yarn.lock ./
-RUN yarn install
-COPY app .
+COPY package.json yarn.lock ./
+COPY app/package.json ./app/package.json
+COPY packages/file-graph/package.json ./packages/file-graph/package.json
+RUN yarn install --frozen-lockfile
+
+COPY app ./app
+COPY packages/file-graph ./packages/file-graph
 RUN yarn build
 
 # Production stage
 FROM nginx:stable-alpine as production-stage
-COPY --from=build-stage /app/dist /usr/share/nginx/html
+COPY --from=build-stage /app/app/dist /usr/share/nginx/html
 COPY deployment/nginx.conf /etc/nginx/conf.d/default.conf
 EXPOSE 80
 CMD ["nginx", "-g", "daemon off;"]
diff --git a/app/Dockerfile.dev b/app/Dockerfile.dev
index 61d9568..a58a1cc 100644
--- a/app/Dockerfile.dev
+++ b/app/Dockerfile.dev
@@ -1,7 +1,10 @@
 FROM node
 
 WORKDIR /app
-COPY app/package.json yarn.lock ./
+COPY package.json yarn.lock ./
+COPY app/package.json ./app/package.json
+COPY packages/file-graph/package.json ./packages/file-graph/package.json
 RUN yarn install --frozen-lockfile
-COPY app .
+
+WORKDIR /app/app
 CMD [ "yarn", "dev:docker", "--host", "0.0.0.0" ]
\ No newline at end of file
diff --git a/app/package.json b/app/package.json
index 8f5147c..4381231 100644
--- a/app/package.json
+++ b/app/package.json
@@ -11,6 +11,7 @@
     "preview": "vite preview"
   },
   "dependencies": {
+    "@linux-odyssey/file-graph": "*",
     "@fortawesome/fontawesome-svg-core": "^6.4.0",
     "@fortawesome/free-brands-svg-icons": "^6.4.0",
     "@fortawesome/free-regular-svg-icons": "^6.4.0",
diff --git a/app/src/components/FileNode.vue b/app/src/components/FileNode.vue
new file mode 100644
index 0000000..ee4636e
--- /dev/null
+++ b/app/src/components/FileNode.vue
@@ -0,0 +1,29 @@
+<template>
+  <a href="#" :class="nodeClass()">{{ node.discovered ? node.name : '???' }}</a>
+  <ul :class="nodeClass()" v-if="node.children && node.children.length > 0">
+    <li :class="nodeClass()" v-for="child in node.children" :key="child.path">
+      <FileNode :node="child" :pwd="pwd" />
+    </li>
+  </ul>
+</template>
+
+<script setup>
+const props = defineProps({
+  node: {
+    type: Object,
+    required: true,
+  },
+  pwd: {
+    type: String,
+    required: true,
+    default: '',
+  },
+})
+
+function nodeClass() {
+  const classes = []
+  if (props.node.discovered) classes.push('discovered')
+  if (props.node.path === props.pwd) classes.push('pwd')
+  return classes
+}
+</script>
diff --git a/app/src/components/QuestPart.vue b/app/src/components/QuestPart.vue
index 933d1d8..8e6a46f 100644
--- a/app/src/components/QuestPart.vue
+++ b/app/src/components/QuestPart.vue
@@ -1,19 +1,19 @@
 <script setup>
-import { ref } from 'vue'
+import { onMounted, ref } from 'vue'
 import api from '../utils/api'
+import sessionManager from '../utils/session'
 
 const questData = ref(null)
 const questErr = ref(null)
 
-api
-  .get('/quests/helloworld')
-  .then((res) => {
-    console.log(res.data)
+onMounted(async () => {
+  try {
+    const res = await api.get('/quests/helloworld')
     questData.value = res.data
-  })
-  .catch((err) => {
+  } catch (err) {
     questErr.value = err
-  })
+  }
+})
 </script>
 
 <template>
@@ -26,8 +26,24 @@ api
     </h1>
     <hr class="border-text-tertiary border-2 my-2" />
     <div>
-      <p class="text-red-500" v-if="questErr">{{ questErr.message }}</p>
-      <p class="text-text" v-else-if="questData">{{ questData.instruction }}</p>
+      <div v-if="questData">
+        <p class="text-text">{{ questData.instruction }}</p>
+        <br />
+        <p class="text-text">Tasks:</p>
+        <ul v-if="sessionManager.session.value">
+          <li v-for="task in sessionManager.session.value.tasks" :key="task.id">
+            <p class="text-text">
+              <input
+                type="checkbox"
+                class="form-checkbox"
+                disabled
+                :checked="task.completed"
+              />{{ task.name }}
+            </p>
+          </li>
+        </ul>
+      </div>
+      <p class="text-red-500" v-else-if="questErr">{{ questErr.message }}</p>
       <p class="text-text" v-else>Loading...</p>
     </div>
   </div>
diff --git a/app/src/components/TreeChart.vue b/app/src/components/TreeChart.vue
index f5ab0e6..93c77b3 100644
--- a/app/src/components/TreeChart.vue
+++ b/app/src/components/TreeChart.vue
@@ -1,56 +1,16 @@
-<script setup></script>
+<script setup>
+import FileNode from './FileNode.vue'
+import sessionManager from '../utils/session.js'
+</script>
 
 <template>
   <div class="tree w-max h-max">
     <ul>
       <li>
-        <a href="#">Parent</a>
-        <ul>
-          <li>
-            <a href="#">Child</a>
-            <ul>
-              <li>
-                <a href="#">Grand Child</a>
-              </li>
-            </ul>
-          </li>
-          <li>
-            <a href="#">Child</a>
-            <ul>
-              <li><a href="#">Grand Child</a></li>
-              <li>
-                <a href="#">Grand Child</a>
-                <ul>
-                  <li>
-                    <a href="#">Great Grand Child</a>
-                  </li>
-                  <li>
-                    <a href="#">Great Grand Child</a>
-                  </li>
-                  <li>
-                    <a href="#">Great Grand Child</a>
-                    <ul>
-                      <li><a>Grand Child</a></li>
-                      <li>
-                        <a>Grand Child</a>
-                        <ul>
-                          <li>
-                            <a>Great Grand Child</a>
-                          </li>
-                          <li>
-                            <a>Great Grand Child</a>
-                          </li>
-                        </ul>
-                      </li>
-                      <li><a>Grand Child</a></li>
-                    </ul>
-                  </li>
-                </ul>
-              </li>
-              <li><a href="#">Grand Child</a></li>
-            </ul>
-          </li>
-        </ul>
+        <FileNode
+          :node="sessionManager.graph.value"
+          :pwd="sessionManager.pwd.value"
+        />
       </li>
     </ul>
   </div>
@@ -149,17 +109,18 @@
   -moz-transition: all 0.5s;
 }
 
-.tree li a:hover,
-.tree li a:hover + ul li a {
+a.discovered {
   background: #c8e4f8;
   color: #000;
   border: 1px solid #94a0b4;
 }
 
-.tree li a:hover + ul li::after,
-.tree li a:hover + ul li::before,
-.tree li a:hover + ul::before,
-.tree li a:hover + ul ul::before {
+ul.discovered,
+li.discovered {
   border-color: #94a0b4;
 }
+
+a.pwd {
+  box-shadow: 0 0 10px rgba(255, 255, 255, 0.5); /* 水平偏移、垂直偏移、模糊半徑、顏色 */
+}
 </style>
diff --git a/app/src/utils/session.js b/app/src/utils/session.js
index 80ab25c..1bf5a21 100644
--- a/app/src/utils/session.js
+++ b/app/src/utils/session.js
@@ -1,10 +1,19 @@
 import { ref } from 'vue'
+import { FileGraph } from '@linux-odyssey/file-graph'
 import api from './api'
 
 class SessionManager {
   constructor(questId) {
     this.questId = questId
     this.session = ref(null)
+    this.graph = ref(
+      new FileGraph({
+        path: '/',
+        type: 'folder',
+        discovered: false,
+      })
+    )
+    this.pwd = ref('')
   }
 
   getSession() {
@@ -14,6 +23,20 @@ class SessionManager {
   setSession(session) {
     console.log('Update session:', session)
     this.session.value = session
+    this.graph.value = new FileGraph(session.graph)
+  }
+
+  handleGraphUpdate(event) {
+    if (event.discover) {
+      this.graph.value.discover(event.discover)
+    }
+    if (event.pwd) {
+      this.pwd.value = event.pwd
+    }
+  }
+
+  setTasks(tasks) {
+    this.session.value.tasks = tasks
   }
 
   async createSession() {
@@ -37,7 +60,8 @@ class SessionManager {
     if (sessions.length === 0) {
       return
     }
-    const session = sessions[sessions.length - 1]
+    const { _id: sessionId } = sessions[sessions.length - 1]
+    const { data: session } = await api.get(`/sessions/${sessionId}`)
     console.log(
       `Last Session ID: ${session._id}, Quest: ${session.quest}, Created At: ${session.createdAt}`
     )
@@ -49,6 +73,7 @@ class SessionManager {
     if (!this.session.value) {
       await this.createSession()
     }
+    console.log(this.session.value)
   }
 }
 
diff --git a/app/src/views/GamePage.vue b/app/src/views/GamePage.vue
index 4aff284..27f152d 100644
--- a/app/src/views/GamePage.vue
+++ b/app/src/views/GamePage.vue
@@ -18,6 +18,12 @@ onMounted(async () => {
   socket.on('disconnect', function close() {
     console.log('Disconnected from the server.')
   })
+  socket.on('graph', (event) => {
+    sessionManager.handleGraphUpdate(event)
+  })
+  socket.on('tasks', (tasks) => {
+    sessionManager.setTasks(tasks)
+  })
   socket.on('message', console.log)
 })
 </script>
diff --git a/docker-compose.prod.yml b/docker-compose.prod.yml
index e448dad..6611c98 100644
--- a/docker-compose.prod.yml
+++ b/docker-compose.prod.yml
@@ -40,6 +40,12 @@ services:
       - MONGO_PASS=user-password
       - MONGO_DB=odyssey-test
 
+  base:
+    image: lancatlin/linux-odyssey:base
+    build:
+      context: .
+      dockerfile: ./quests/Dockerfile
+
   helloworld:
     image: lancatlin/linux-odyssey:helloworld
     build:
diff --git a/docker-compose.yml b/docker-compose.yml
index e5b2474..1a637f5 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -9,7 +9,8 @@ services:
     networks:
       - internal
     volumes:
-      - ./app/src:/app/src
+      - ./app:/app/app
+      - ./packages/file-graph:/app/packages/file-graph
   backend:
     build:
       context: .
@@ -21,8 +22,9 @@ services:
       - '3000:3000'
     volumes:
       - /var/run/docker.sock:/var/run/docker.sock
-      - ./server/src:/app/src
-      - ./server/swagger.yaml:/app/swagger.yaml
+      - ./server/src:/app/server/src
+      - ./server/swagger.yaml:/app/server/swagger.yaml
+      - ./packages:/app/packages
       - ./quests:/app/quests
     environment:
       - MONGO_URL=mongodb://db:27017/odyssey-test
@@ -46,6 +48,12 @@ services:
     environment:
       - SWAGGER_FILE=/tmp/swagger.yaml
 
+  base:
+    image: lancatlin/linux-odyssey:base
+    build:
+      context: .
+      dockerfile: ./quests/Dockerfile
+
   helloworld:
     image: lancatlin/linux-odyssey:helloworld
     build:
diff --git a/package.json b/package.json
index 3713418..cc4fd76 100644
--- a/package.json
+++ b/package.json
@@ -13,10 +13,12 @@
   "scripts": {
     "app": "yarn workspace app dev",
     "app:local": "yarn workspace app dev:local",
+    "build": "yarn workspace app build",
     "server": "yarn workspace server dev",
     "cli": "yarn workspace client start",
     "lint": "eslint --ext .js,.vue .",
     "format": "prettier --ignore-path .prettierignore --write app server",
+    "test": "yarn node --experimental-vm-modules $(yarn bin jest)",
     "clean": "yarn workspace server clean"
   },
   "devDependencies": {
diff --git a/packages/client/index.js b/packages/client/index.js
index 5363631..99ec410 100644
--- a/packages/client/index.js
+++ b/packages/client/index.js
@@ -1,8 +1,11 @@
 /* eslint-disable no-underscore-dangle */
+import axios from 'axios'
 import { stdin, stdout, exit } from 'process'
 import { io } from 'socket.io-client'
 import { program } from 'commander'
 
+let api = null
+
 function get(key, defaultValue) {
   const value = process.env[key]
   if (value != null) {
@@ -14,26 +17,29 @@ stdin.setRawMode(true)
 stdin.resume()
 stdin.setEncoding('utf8')
 
-async function createdSession() {
+function debug(...args) {
+  if (program.opts().debug) {
+    console.debug(...args)
+  }
+}
+
+async function createSession() {
   console.log('Creating a new session...')
-  const res = await fetch(`${program.opts().host}/api/v1/sessions`, {
-    method: 'POST',
-    headers: {
-      'Content-Type': 'application/json',
-    },
-    body: JSON.stringify({
-      quest_id: 'helloworld',
-    }),
+  const res = await api.post('/sessions', {
+    quest_id: 'helloworld',
   })
-  const data = await res.json()
+  const { data } = await res
+  if (!data._id) {
+    console.error(data)
+    exit(1)
+  }
   console.log('Created Session ID:', data._id)
   return data
 }
 
 async function getSessionList() {
-  const res = await fetch(`${program.opts().host}/api/v1/sessions`)
-  const data = await res.json()
-  return data
+  const res = await api.get('/sessions')
+  return res.data
 }
 
 async function lastSession() {
@@ -73,8 +79,11 @@ async function connect(sessionId) {
   })
 
   socket.on('graph', (data) => {
-    console.log('receive graph:')
-    console.log(data)
+    debug('receive graph:', data)
+  })
+
+  socket.on('tasks', (data) => {
+    debug('receive tasks:', data)
   })
 
   socket.on('close', function close() {
@@ -93,12 +102,18 @@ async function connect(sessionId) {
 }
 
 async function main() {
+  api = axios.create({
+    baseURL: `${program.opts().host}/api/v1`,
+    headers: {
+      'Content-Type': 'application/json',
+    },
+  })
   console.log(program.opts().session)
   // main()
 
   const sessionId =
     program.opts().session ||
-    (program.opts().create ? (await createdSession())._id : null) ||
+    (program.opts().create ? (await createSession())._id : null) ||
     (await lastSession())._id
 
   await connect(sessionId)
@@ -107,6 +122,7 @@ async function main() {
 program
   .option('-s, --session <string>', 'Session ID')
   .option('-c, --create', 'Create a new session')
+  .option('-d, --debug', 'Debug mode')
   .option(
     '-h, --host <string>',
     'Server host',
diff --git a/packages/client/package.json b/packages/client/package.json
index 968cf60..983b984 100644
--- a/packages/client/package.json
+++ b/packages/client/package.json
@@ -8,6 +8,7 @@
     "start": "node index.js"
   },
   "dependencies": {
+    "axios": "^1.5.0",
     "commander": "^11.0.0",
     "socket.io-client": "^4.7.2"
   }
diff --git a/packages/container/print.js b/packages/container/print.js
index 76a77a5..e2b6f95 100644
--- a/packages/container/print.js
+++ b/packages/container/print.js
@@ -36,6 +36,12 @@ function waitForEnter() {
   })
 }
 
+function waitTime(time) {
+  return new Promise((resolve) => {
+    setTimeout(resolve, time)
+  })
+}
+
 function printLine(content, delay, color) {
   const printContent = `${colorize(content, color)}${RETURN_SYMBOL}`
   return new Promise((resolve) => {
@@ -59,6 +65,9 @@ function printLine(content, delay, color) {
 }
 
 async function printResponses(responses, delay = 100) {
+  await waitTime(1000)
+  console.log()
+  console.log(colorize('-------------', 'white'))
   for (const response of responses) {
     const { type, content, speaker, color } = response
     if (type === 'narrative') {
diff --git a/packages/file-graph/file.js b/packages/file-graph/file.js
index 986e280..983cbeb 100644
--- a/packages/file-graph/file.js
+++ b/packages/file-graph/file.js
@@ -2,7 +2,7 @@ import { basename } from './utils.js'
 
 export default class File {
   constructor({ path, type, discovered }) {
-    this.path = path
+    this.path = path.trimEnd('/')
     this.name = basename(path)
     this.type = type
     this.discovered = discovered
@@ -13,6 +13,9 @@ export default class File {
   }
 
   contains(file) {
-    return this.isDirectory() && file.path.startsWith(this.path)
+    if (!this.isDirectory()) return false
+    if (this.path === file.path) return true
+    const parentPath = this.path.endsWith('/') ? this.path : `${this.path}/`
+    return file.path.startsWith(parentPath)
   }
 }
diff --git a/packages/file-graph/file.test.js b/packages/file-graph/file.test.js
new file mode 100644
index 0000000..175ac73
--- /dev/null
+++ b/packages/file-graph/file.test.js
@@ -0,0 +1,49 @@
+import File from './file.js'
+
+describe('File', () => {
+  let file
+  beforeEach(() => {
+    file = new File({ path: '/home/user', type: 'folder', discovered: true })
+  })
+
+  const filesToCompare = [
+    {
+      name: 'direct children',
+      path: '/home/user/file1.txt',
+      answer: true,
+    },
+    {
+      name: 'nested children',
+      path: '/home/user/folder1/file1.txt',
+      answer: true,
+    },
+    {
+      name: 'outer file',
+      path: '/etc/file1.txt',
+      answer: false,
+    },
+    {
+      name: 'fake children',
+      path: '/home/user2/file2.txt',
+      answer: false,
+    },
+    {
+      name: 'same',
+      path: '/home/user',
+      answer: true,
+    },
+    {
+      name: 'same with trailing',
+      path: '/home/user/',
+      answer: true,
+    },
+  ]
+
+  filesToCompare.forEach(({ name, path, answer }) => {
+    test(`${name}: ${path}`, () => {
+      const fileToCompare = new File({ path, type: 'file', discovered: true })
+      console.log(fileToCompare)
+      expect(file.contains(fileToCompare)).toBe(answer)
+    })
+  })
+})
diff --git a/packages/file-graph/fileGraph.js b/packages/file-graph/fileGraph.js
index 611fc4c..a2a924e 100644
--- a/packages/file-graph/fileGraph.js
+++ b/packages/file-graph/fileGraph.js
@@ -1,17 +1,13 @@
 import FileNode from './fileNode.js'
 
-export default class FileGraph {
-  constructor(root) {
-    this.root = new FileNode(root)
-  }
-
+export default class FileGraph extends FileNode {
   add(files) {
     // Logic to add the file node to the graph
     // You can implement the logic to find the appropriate location in the file graph
     // If the file node already exists, throw an error
 
     files.forEach((file) => {
-      this.root.addChild(file, { makeParents: true })
+      this.addChild(file, { makeParents: true })
     })
   }
 
@@ -21,7 +17,7 @@ export default class FileGraph {
     // If the file node doesn't exist, throw an error
 
     files.forEach((file) => {
-      this.root.removeChild(file)
+      this.removeChild(file)
     })
   }
 
@@ -36,7 +32,7 @@ export default class FileGraph {
       newNode.addChild(file, { makeParents: true })
     })
 
-    this.root.merge(newNode)
+    this.merge(newNode)
   }
 
   handleEvent(event) {
diff --git a/packages/file-graph/fileGraph.test.js b/packages/file-graph/fileGraph.test.js
index 7f6447a..cff996a 100644
--- a/packages/file-graph/fileGraph.test.js
+++ b/packages/file-graph/fileGraph.test.js
@@ -30,9 +30,9 @@ describe('FileGraph', () => {
 
     fileGraph.add(filesToAdd)
 
-    console.log(fileGraph.root.toString())
-    expect(fileGraph.root.children.length).toBe(1)
-    expect(fileGraph.root.children[0].children[0].children.length).toBe(2)
+    console.log(fileGraph.toString())
+    expect(fileGraph.children.length).toBe(1)
+    expect(fileGraph.children[0].children[0].children.length).toBe(2)
   })
 
   test('remove files from graph', () => {
@@ -76,14 +76,44 @@ describe('FileGraph', () => {
 
     fileGraph.remove(filesToRemove)
 
-    console.log(fileGraph.root.toString())
-    expect(fileGraph.root.children[0].children[0].children.length).toBe(1)
-    expect(fileGraph.root.children[0].children[0].children[0].name).toBe(
-      'folder1'
+    console.log(fileGraph.toString())
+    expect(fileGraph.children[0].children[0].children.length).toBe(1)
+    expect(fileGraph.children[0].children[0].children[0].name).toBe('folder1')
+    expect(fileGraph.children[0].children[0].children[0].children.length).toBe(
+      0
+    )
+  })
+
+  test('discover', () => {
+    const fileGraph = new FileGraph(root)
+    const discoverFiles = [
+      {
+        path: '/home/user/folder1',
+        type: 'folder',
+        discovered: true,
+      },
+      {
+        path: '/home/user/folder1/file1.txt',
+        type: 'file',
+        discovered: true,
+      },
+      {
+        path: '/home/user/folder1/file2.txt',
+        type: 'file',
+        discovered: true,
+      },
+      {
+        path: '/home/user/folder1/file3.txt',
+        type: 'file',
+        discovered: true,
+      },
+    ]
+
+    fileGraph.discover(discoverFiles)
+    console.log(fileGraph.toString())
+    expect(fileGraph.children[0].children[0].children[0].children.length).toBe(
+      3
     )
-    expect(
-      fileGraph.root.children[0].children[0].children[0].children.length
-    ).toBe(0)
   })
 
   test('discover new files', () => {
@@ -113,7 +143,7 @@ describe('FileGraph', () => {
     ]
 
     fileGraph.add(filesToAdd)
-    console.log(fileGraph.root.toString())
+    console.log(fileGraph.toString())
 
     const newFiles = [
       {
@@ -140,10 +170,10 @@ describe('FileGraph', () => {
 
     fileGraph.discover(newFiles)
 
-    console.log(fileGraph.root.toString())
-    expect(fileGraph.root.children[0].children[0].children.length).toBe(2)
-    expect(
-      fileGraph.root.children[0].children[0].children[0].children.length
-    ).toBe(3)
+    console.log(fileGraph.toString())
+    expect(fileGraph.children[0].children[0].children.length).toBe(2)
+    expect(fileGraph.children[0].children[0].children[0].children.length).toBe(
+      3
+    )
   })
 })
diff --git a/packages/file-graph/fileNode.js b/packages/file-graph/fileNode.js
index 96fc60c..cf8f089 100644
--- a/packages/file-graph/fileNode.js
+++ b/packages/file-graph/fileNode.js
@@ -9,7 +9,8 @@ import { basename, dirname } from './utils.js'
 export default class FileNode extends File {
   constructor(file) {
     super(file)
-    this.children = file.children || []
+    this.children = []
+    if (file.children) this.children = file.children.map((f) => new FileNode(f))
   }
 
   addChild(fileInput, { makeParents } = {}) {
@@ -76,7 +77,12 @@ export default class FileNode extends File {
     // Merge the children of the provided file node into this node
     // If a child already exists, merge the children of that child
     // If a child does not exist, add it to this node
+
+    // If this is the same node, merge the children
     if (fileNode.path === this.path) {
+      // Update the state of this node
+      this.discovered = this.discovered || fileNode.discovered
+
       fileNode.children.forEach((child) => {
         const existingChild = this.children.find((c) => c.path === child.path)
         if (existingChild) {
@@ -92,12 +98,17 @@ export default class FileNode extends File {
           fileNode.children.some((c) => c.path === child.path)
         )
       }
-    } else if (this.contains(fileNode)) {
+    }
+    // If this contains the parent node
+    else if (this.contains(fileNode)) {
       const childNode = this.children.find((child) => child.contains(fileNode))
+      // If a child node exists, merge the provided node into the child node
       if (childNode) {
         childNode.merge(fileNode)
-      } else {
-        this.addChild(fileNode)
+      }
+      // If a child node does not exist, add the provided node as a child
+      else {
+        this.addChild(fileNode, { makeParents: true })
       }
     } else {
       throw new Error(
@@ -108,12 +119,15 @@ export default class FileNode extends File {
     }
   }
 
-  toString() {
-    const indent = '    '.repeat(this.path.split('/').length - 1)
-    const suffix = this.isDirectory() ? '/' : ''
+  toString(level = 0) {
+    let { name } = this
+    const indent = '    '.repeat(level)
+    name = `${indent}${name}`
+    if (this.isDirectory()) name += '/'
+    if (!this.discovered) name += `*`
     const childrenString = this.children
-      .map((child) => child.toString())
+      .map((child) => child.toString(level + 1))
       .join('')
-    return `${indent}${this.name}${suffix}\n${childrenString}`
+    return `${name}\n${childrenString}`
   }
 }
diff --git a/packages/file-graph/fileNode.test.js b/packages/file-graph/fileNode.test.js
index b58e38f..3413dc5 100644
--- a/packages/file-graph/fileNode.test.js
+++ b/packages/file-graph/fileNode.test.js
@@ -193,6 +193,35 @@ describe('FileNode', () => {
     mergeTest(files1, files2, expected)
   })
 
+  test('merge-hidden', () => {
+    const root = {
+      path: '/',
+      type: 'folder',
+    }
+    const files1 = [
+      root,
+      { path: '/a', type: 'file', discovered: true },
+      { path: '/.b', type: 'file', discovered: false },
+      { path: '/c', type: 'folder', discovered: true },
+      { path: '/c/.d', type: 'file', discovered: false },
+    ]
+
+    const files2 = [
+      { path: '/c', type: 'folder', discovered: true },
+      { path: '/c/.d', type: 'file', discovered: true },
+    ]
+
+    const expected = [
+      root,
+      { path: '/a', type: 'file', discovered: true },
+      { path: '/.b', type: 'file', discovered: false },
+      { path: '/c', type: 'folder', discovered: true },
+      { path: '/c/.d', type: 'file', discovered: true },
+    ]
+
+    mergeTest(files1, files2, expected)
+  })
+
   test('remove-child', () => {
     const root = {
       path: '/',
@@ -219,3 +248,39 @@ describe('FileNode', () => {
     removeChildTest(files, fileToRemove, expected)
   })
 })
+
+test('load nested fileNode', () => {
+  const data = {
+    path: '/',
+    type: 'folder',
+    children: [
+      {
+        path: '/a',
+        type: 'file',
+      },
+      {
+        path: '/b',
+        type: 'folder',
+        children: [
+          {
+            path: '/b/c',
+            type: 'folder',
+            children: [
+              {
+                path: '/b/c/d',
+                type: 'file',
+              },
+            ],
+          },
+        ],
+      },
+    ],
+  }
+
+  const node = new FileNode(data)
+
+  console.log(node.toString())
+  expect(node.children.length).toBe(2)
+  expect(node.children[1].children.length).toBe(1)
+  expect(node.children[1].children[0].children.length).toBe(1)
+})
diff --git a/packages/file-graph/index.js b/packages/file-graph/index.js
new file mode 100644
index 0000000..4b1fcda
--- /dev/null
+++ b/packages/file-graph/index.js
@@ -0,0 +1,3 @@
+export { default as File } from './file.js'
+export { default as FileNode } from './fileNode.js'
+export { default as FileGraph } from './fileGraph.js'
diff --git a/packages/file-graph/package.json b/packages/file-graph/package.json
index 69b2535..2388c78 100644
--- a/packages/file-graph/package.json
+++ b/packages/file-graph/package.json
@@ -1,10 +1,9 @@
 {
-  "name": "file-graph",
+  "name": "@linux-odyssey/file-graph",
   "version": "1.0.0",
   "main": "index.js",
+  "exports": "./index.js",
   "license": "MIT",
   "type": "module",
-  "scripts": {
-    "test": "yarn node --experimental-vm-modules $(yarn bin jest)"
-  }
+  "scripts": {}
 }
diff --git a/quests/Dockerfile b/quests/Dockerfile
new file mode 100644
index 0000000..3971f8f
--- /dev/null
+++ b/quests/Dockerfile
@@ -0,0 +1,41 @@
+FROM ubuntu:22.04
+
+LABEL maintainer="wancat@wancat.cc"
+
+RUN apt-get update && apt-get install -y \
+  locales \
+  curl \
+  vim \
+  zsh \
+  ca-certificates \
+  gnupg
+
+RUN sed -i -e 's/# zh_TW.UTF-8 UTF-8/zh_TW.UTF-8 UTF-8/' /etc/locale.gen && \
+  dpkg-reconfigure --frontend=noninteractive locales && \
+  update-locale LANG=zh_TW.UTF-8
+
+ENV LANG zh_TW.UTF-8
+ENV LANGUAGE zh_TW:zh
+ENV LC_ALL zh_TW.UTF-8
+
+ENV NODE_MAJOR=20
+RUN curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg
+RUN echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_$NODE_MAJOR.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list
+RUN apt-get update && apt-get install nodejs -y
+
+WORKDIR /usr/local/lib/container
+
+RUN npm install -g yarn
+COPY ./packages/container/package.json ./yarn.lock ./
+RUN yarn install --modules-folder /usr/src/node_modules
+ENV NODE_PATH=/usr/src/node_modules
+COPY ./packages/container /usr/local/lib/container
+
+ENV USERNAME=commander
+
+RUN adduser --disabled-password --gecos '' --shell /bin/zsh ${USERNAME}
+
+USER ${USERNAME}
+
+WORKDIR /home/${USERNAME}
+
diff --git a/quests/helloworld/Dockerfile b/quests/helloworld/Dockerfile
index 10e45fa..e845379 100644
--- a/quests/helloworld/Dockerfile
+++ b/quests/helloworld/Dockerfile
@@ -1,35 +1,5 @@
-FROM node:latest
+FROM lancatlin/linux-odyssey:base
 
-RUN apt-get update && apt-get install -y \
-  locales \
-  curl \
-  vim \
-  zsh 
-
-RUN sed -i -e 's/# zh_TW.UTF-8 UTF-8/zh_TW.UTF-8 UTF-8/' /etc/locale.gen && \
-  dpkg-reconfigure --frontend=noninteractive locales && \
-  update-locale LANG=zh_TW.UTF-8
-
-ENV LANG zh_TW.UTF-8
-ENV LANGUAGE zh_TW:zh
-ENV LC_ALL zh_TW.UTF-8
-
-WORKDIR /usr/local/lib/container
-
-COPY ./packages/container/package.json ./yarn.lock ./
-
-RUN yarn install --modules-folder /usr/src/node_modules
-
-ENV NODE_PATH=/usr/src/node_modules
-
-COPY ./packages/container /usr/local/lib/container
-
-RUN adduser --disabled-password --gecos '' --shell /bin/zsh rudeus
-
-USER rudeus
-
-WORKDIR /home/rudeus
-
-COPY ./quests/helloworld/home /home/rudeus/
+COPY ./quests/helloworld/home ${HOME}
 
 CMD [ "/bin/zsh" ]
diff --git a/quests/helloworld/game.yml b/quests/helloworld/game.yml
index 2dca81c..c474213 100644
--- a/quests/helloworld/game.yml
+++ b/quests/helloworld/game.yml
@@ -4,11 +4,14 @@ instruction: |
   你身在一個漆黑的房間裡，沒有任何的光線或聲音。你想不起來為什麼自己會來到這裡。但你的腦中隱約感到某個聲音跟你說：
 
   「輸入 echo Hello World!」
+
 stages:
   - id: helloworld
+    task: 使用 `echo Hello World!` 召喚精靈
+    repeatable: false
     condition:
       command:
-        - echo Hello World!
+        - echo Hello World
     responses:
       - type: narrative
         content:
@@ -22,12 +25,16 @@ stages:
           - 由於長久以來依賴於薩加，世人已失去了「指令」的力量，即將走向衰敗和滅亡的命運。
           - 你的任務，就是找回被世人遺忘的指令，奪回 Linux 之根，成為 Linux 之王。
           - 首先，你必須要學會如何使用指令。
-          - 一個指令是由一個指令名稱和一些參數所組成的，而指令名稱和參數之間是以空白字元分隔。
-          - 例如，你可以輸入 `ls` 來查看目前的目錄下有哪些檔案。
+          - 你要學習的第一個指令，是「發現」的指令：`ls`。ls 可以用來查看一個目錄中有哪些檔案。
+          - 現在，請你輸入 `ls` 來查看目前的目錄下有哪些檔案。
     hints:
       - 使用 `ls` 指令來查看目前的目錄下有哪些檔案。
-    next: ls
+
   - id: ls
+    task: 搜索卷軸
+    repeatable: false
+    requirements:
+      - helloworld
     condition:
       command:
         - ls
@@ -47,8 +54,27 @@ stages:
     hints:
       - 使用 `cat [檔案名稱]` 指令來查看檔案內容。
       - 使用 `Tab` 鍵來取得指令提示。
-    next: cat
+
+  - id: escape
+    task: 移動到下一個房間
+    repeatable: false
+    requirements:
+      - ls
+    condition:
+      pwd:
+        - '^/home/commander/jail$'
+    responses:
+      - type: narrative
+        content:
+          - 你進入了一個牢房，牢房裡有一個門，門上有一個鎖。
+          - 突然，你聽到門上鎖的聲音。
+          - 轉身發現，自己被關在了牢房裡。
+
   - id: cat
+    task: 查看卷軸
+    repeatable: false
+    requirements:
+      - ls
     condition:
       command:
         - cat forgotten_scroll.txt
@@ -58,8 +84,12 @@ stages:
         content:
           - 你找到的是一個被遺忘的卷軸，上面寫著解除封印的咒語。
           - 使用你已經知道的 `echo` 指令來詠唱咒語。
-    next: echo
-  - id: echo
+
+  - id: END
+    task: 解除封印
+    repeatable: false
+    requirements:
+      - cat
     condition:
       command:
         - echo Unix, rise and guide me!
@@ -71,5 +101,4 @@ stages:
         speaker: Unix Spirit
         content:
           - 恭喜您，我的指令官，您已成功解除了封印。
-          - 請跟著在接下了關卡，一步步取回 Linux 之根吧！
-    next: END
+          - 請跟著在接下來的關卡，一步步取回 Linux 之根吧！
diff --git a/quests/helloworld/story.md b/quests/helloworld/story.md
index 66a2a21..528cbf8 100644
--- a/quests/helloworld/story.md
+++ b/quests/helloworld/story.md
@@ -6,7 +6,10 @@
 
 1. echo 朗誦魔術
 2. ls 搜索魔術
-3. cat 檢察魔術
+3. cat 檢查魔術
+4. cd 移動魔術
+5. mv 搬運魔術
+6. rm 消滅魔術
 
 ## 關卡設計
 
@@ -14,8 +17,8 @@
 2. 學習到了 ls
 3. 使用 ls 找到地上放置了一個卷軸
 4. 學習 cat
-5. 使用 cat 查看卷軸內容
-6. 用 echo 念出卷軸上的咒語
+5. 使用 cat 查看卷軸內容，發現
+6.
 
 請幫我設計：
 
diff --git a/server/Dockerfile b/server/Dockerfile
index 6ee9ddd..078876b 100644
--- a/server/Dockerfile
+++ b/server/Dockerfile
@@ -1,13 +1,17 @@
 FROM node
 
 WORKDIR /app
-COPY server/package.json yarn.lock ./
+COPY ./package.json yarn.lock ./
+COPY server/package.json ./server/package.json
+COPY packages/file-graph/package.json ./packages/file-graph/package.json
 RUN yarn install --frozen-lockfile --production
 
-COPY server/src ./src
-COPY server/swagger.yaml .
+COPY ./server ./server
+COPY packages/file-graph ./packages/file-graph
 COPY quests ./quests
 
+WORKDIR /app/server
+
 ENV NODE_ENV=production
 ENV HOST=0.0.0.0
 ENV PORT=3000
diff --git a/server/Dockerfile.dev b/server/Dockerfile.dev
index 86186c2..738d4ea 100644
--- a/server/Dockerfile.dev
+++ b/server/Dockerfile.dev
@@ -1,7 +1,9 @@
 FROM node
 
 WORKDIR /app
-COPY server/package.json yarn.lock ./
+COPY package.json yarn.lock ./
+COPY server/package.json ./server/package.json
+COPY packages/file-graph/package.json ./packages/file-graph/package.json
 RUN yarn install --frozen-lockfile
 
 ENV NODE_ENV=development
@@ -9,4 +11,4 @@ ENV HOST=0.0.0.0
 ENV PORT=3000
 EXPOSE 3000
 
-CMD [ "yarn", "dev" ]
\ No newline at end of file
+CMD [ "yarn", "server" ]
\ No newline at end of file
diff --git a/server/package.json b/server/package.json
index 0fe6b1f..678672b 100644
--- a/server/package.json
+++ b/server/package.json
@@ -10,6 +10,7 @@
     "clean": "node src/cli/clean.js"
   },
   "dependencies": {
+    "@linux-odyssey/file-graph": "*",
     "commander": "^11.0.0",
     "dockerode": "^3.3.5",
     "express": "^4.18.2",
diff --git a/server/src/api/controllers/commandController.js b/server/src/api/controllers/commandController.js
index 5526308..fd583e1 100644
--- a/server/src/api/controllers/commandController.js
+++ b/server/src/api/controllers/commandController.js
@@ -2,12 +2,11 @@
 import Command from '../../models/command.js'
 import Session from '../../models/session.js'
 import { verifySessionJWT } from '../../utils/auth.js'
-import commandHandler from '../../game/commandHandler.js'
+import CommandHandler from '../../game/commandHandler.js'
 
 export async function newCommand(req, res) {
   console.log('new command:', req.body)
   const { token, command, pwd, output, error, ...additionalData } = req.body
-  console.log('additionalData:', additionalData)
 
   if (!command) {
     res.status(400).json({ message: 'command is required' })
@@ -36,7 +35,6 @@ export async function newCommand(req, res) {
   }
 
   session.lastActivityAt = new Date()
-  await session.save()
 
   const c = new Command({
     session,
@@ -48,8 +46,10 @@ export async function newCommand(req, res) {
 
   await c.save()
 
-  const response = await commandHandler(session, c, additionalData)
-  console.log(response)
+  const commandHandler = new CommandHandler(session, c, additionalData)
+
+  const response = await commandHandler.run()
 
   res.status(201).json(response)
+  await session.save()
 }
diff --git a/server/src/api/controllers/sessionController.js b/server/src/api/controllers/sessionController.js
index aa37963..3130288 100644
--- a/server/src/api/controllers/sessionController.js
+++ b/server/src/api/controllers/sessionController.js
@@ -1,6 +1,7 @@
 import Session from '../../models/session.js'
 import Quest from '../../models/quest.js'
 import { createContainer } from '../../containers/docker.js'
+import SessionHandler from '../../game/sessionHandler.js'
 
 export async function getSessionList(req, res) {
   try {
@@ -45,20 +46,22 @@ export async function createSession(req, res) {
   }
 
   try {
-    const progress = quest.stages[0].id
     const container = await createContainer(
       `quest-${quest.id}-${req.user.username}-${Date.now()}`
     )
     const newSession = new Session({
       user: req.user,
       quest,
-      progress,
       containerId: container.id,
     })
 
-    const session = await newSession.save()
-    res.status(201).json(session)
+    const session = new SessionHandler(newSession)
+    session.addNewTasks()
+
+    await newSession.save()
+    res.status(201).json(newSession)
   } catch (err) {
+    console.error(err)
     res.status(400).json({ message: err.message })
   }
 }
diff --git a/server/src/api/socket.js b/server/src/api/socket.js
index 25d26d6..67b9d77 100644
--- a/server/src/api/socket.js
+++ b/server/src/api/socket.js
@@ -5,19 +5,85 @@ import { defaultUser, genSessionJWT } from '../utils/auth.js'
 
 const sessions = new Map()
 
-function listenToSession(sessionId, event, callback) {
-  const callbacks = sessions.get((sessionId, event)) || []
-  sessions.set((sessionId, event), callbacks.concat(callback))
+function listenToSession(sessionId, callback) {
+  const callbacks = sessions.get(sessionId) || []
+  sessions.set(sessionId, callbacks.concat(callback))
 }
 
-export function pushToSession(sessionId, event, data) {
-  console.log('Push:', sessionId, event)
-  const callbacks = sessions.get((sessionId, event))
+function removeFromSession(sessionId, event, callback) {
+  const callbacks = sessions.get((sessionId, event)) || []
+  sessions.set(
+    (sessionId, event),
+    callbacks.filter((cb) => cb !== callback)
+  )
+}
+
+export function pushToSession(sessionId, event, ...args) {
+  const callbacks = sessions.get(sessionId)
   if (callbacks) {
-    callbacks.forEach((callback) => callback(data))
+    callbacks.forEach((callback) => callback(event, ...args))
   }
 }
 
+async function connectClient(socket) {
+  console.log('Connected to the client.')
+  const sessionId = socket.handshake.query.session_id
+  if (!sessionId) {
+    socket.send('Session ID not found.')
+    socket.disconnect()
+    return
+  }
+
+  const session = await Session.findOne({
+    _id: sessionId,
+    user: socket.user,
+    status: 'active',
+  })
+  if (!session) {
+    socket.send('Session not found.')
+    socket.disconnect()
+    return
+  }
+
+  const token = await genSessionJWT(session)
+
+  let container
+  let stream
+  try {
+    container = await getAndStartContainer(session.containerId)
+    stream = await attachContainer(container, { token })
+  } catch (err) {
+    socket.send(err.message)
+    socket.disconnect()
+    return
+  }
+
+  stream.socket.on('data', (chunk) => {
+    socket.emit('terminal', chunk.toString())
+  })
+
+  socket.on('message', console.log)
+
+  socket.on('terminal', function incoming(message) {
+    stream.socket.write(message)
+  })
+
+  const socketCallback = (event, ...data) => {
+    socket.emit(event, ...data)
+  }
+
+  listenToSession(session.id, socketCallback)
+
+  socket.on('disconnect', () => {
+    console.log('Disconnected from the client.')
+    stream.socket.write('exit\n')
+    removeFromSession(session.id, socketCallback)
+    stream.destroy()
+  })
+
+  socket.send(`${container.id}\n`)
+}
+
 export default (server) => {
   const io = new Server(server)
 
@@ -34,55 +100,5 @@ export default (server) => {
     }
   })
 
-  io.on('connection', async (socket) => {
-    console.log('Connected to the client.')
-    const sessionId = socket.handshake.query.session_id
-    if (!sessionId) {
-      socket.send('Session ID not found.')
-      socket.disconnect()
-      return
-    }
-    let session
-    try {
-      session = await Session.findOne({
-        _id: sessionId,
-        user: socket.user,
-        status: 'active',
-      })
-      if (!session) {
-        socket.send('Session not found.')
-        socket.disconnect()
-        return
-      }
-    } catch (err) {
-      socket.send(err.message)
-      socket.disconnect()
-      return
-    }
-
-    const token = await genSessionJWT(session)
-
-    const container = await getAndStartContainer(session.containerId)
-    const stream = await attachContainer(container, { token })
-
-    stream.on('data', (chunk) => {
-      socket.emit('terminal', chunk.toString())
-    })
-
-    socket.on('message', console.log)
-
-    socket.on('terminal', function incoming(message) {
-      stream.write(message)
-    })
-
-    socket.on('close', () => {
-      console.log('Disconnected from the client.')
-    })
-
-    listenToSession(session.id, 'graph', (data) => {
-      socket.emit('graph', data)
-    })
-
-    socket.send(`${container.id}\n`)
-  })
+  io.on('connection', connectClient)
 }
diff --git a/server/src/containers/docker.js b/server/src/containers/docker.js
index b687897..a4c768f 100644
--- a/server/src/containers/docker.js
+++ b/server/src/containers/docker.js
@@ -15,7 +15,7 @@ const containerOptions = {
     Binds:
       !config.isProduction && config.hostPwd
         ? [
-            `${config.hostPwd}/quests/helloworld/home:/home/rudeus`,
+            `${config.hostPwd}/quests/helloworld/home:/home/commander`,
             `${config.hostPwd}/packages/container:/usr/local/lib/container`,
           ]
         : [],
@@ -35,16 +35,11 @@ export async function createContainer(name) {
 
 export async function getAndStartContainer(id) {
   console.log(`Getting container: ${id}`)
-  let container
-  try {
-    container = engine.getContainer(id)
-    if (!(await container.inspect()).State.Running) {
-      await container.start()
-    }
-    return container
-  } catch (error) {
-    throw new Error(`Container not found: ${id}`)
+  const container = engine.getContainer(id)
+  if (!(await container.inspect()).State.Running) {
+    await container.start()
   }
+  return container
 }
 
 export async function attachContainer(container, { token }) {
@@ -63,7 +58,7 @@ export async function attachContainer(container, { token }) {
     Tty: true,
   })
 
-  return execOutput.socket
+  return execOutput
 }
 
 export async function deleteContainer(id) {
diff --git a/server/src/game/commandHandler.js b/server/src/game/commandHandler.js
index 35335f3..f18900c 100644
--- a/server/src/game/commandHandler.js
+++ b/server/src/game/commandHandler.js
@@ -1,58 +1,83 @@
 import minimist from 'minimist'
+import { FileGraph } from '@linux-odyssey/file-graph'
 import Quest from '../models/quest.js'
 import { pushToSession } from '../api/socket.js'
+import SessionHandler from './sessionHandler.js'
 
 // 檢查 pattern 是否符合 input
 const checkMatch = (pattern, input) => {
-  if (pattern.length === 0) return true
+  if (!pattern || pattern.length === 0) return true
   return pattern.some((p) => {
     const regex = new RegExp(p)
     return regex.test(input)
   })
 }
 
-async function commandHandler(session, commandInput, additionalData) {
-  const quest = await Quest.findById(session.quest)
-  const stage = quest.stages.find((s) => s.id === session.progress)
-  if (!stage) {
-    console.error('stage not found', session.progress)
-    return {}
+export default class CommandHandler extends SessionHandler {
+  constructor(session, commandInput, additionalData) {
+    super(session)
+    this.commandInput = commandInput
+    this.argv = minimist(this.commandInput.command.split(' '))
+
+    this.additionalData = additionalData
   }
 
-  const argv = minimist(commandInput.command.split(' '))
-
-  if ('discover' in additionalData) {
-    pushToSession(session.id, 'graph', {
-      pwd: commandInput.pwd,
-      discover: additionalData.discover,
-    })
-  } else if (argv._[0] === 'cd') {
-    pushToSession(session.id, 'graph', { pwd: commandInput.pwd })
+  handleEvent() {
+    if (this.additionalData.discover) this.discoverHandler()
   }
 
-  const commandMatch = checkMatch(stage.condition.command, commandInput.command)
-  const outputMatch = checkMatch(stage.condition.output, commandInput.output)
-  const errorMatch = checkMatch(stage.condition.error, commandInput.output)
+  handleCommand() {
+    const command = this.argv._[0]
+    switch (command) {
+      case 'cd':
+        pushToSession(this.session.id, 'graph', {
+          pwd: this.commandInput.pwd,
+        })
+        break
 
-  if (!commandMatch || !outputMatch || !errorMatch) return {}
+      case 'ls':
+        pushToSession(this.session.id, 'graph', {
+          pwd: this.commandInput.pwd,
+          discover: this.additionalData.discover,
+        })
+        break
 
-  session.completion.push(session.progress)
-  session.hints.push(...stage.hints)
-
-  session.progress = stage.next
-  await session.save()
-
-  if (stage.next === 'END') {
-    session.status = 'finished'
-    session.finishedAt = new Date()
-    await session.save()
+      default:
+        break
+    }
   }
 
-  return {
-    responses: stage.responses,
-    hints: stage.hints,
-    end: stage.next === 'END',
+  async discoverHandler() {
+    const graph = new FileGraph(this.session.graph)
+    graph.discover(this.additionalData.discover)
+    this.session.graph = graph
+  }
+
+  isMatch(stage) {
+    const keys = ['command', 'output', 'error', 'pwd']
+    return keys.every((k) =>
+      checkMatch(stage.condition[k], this.commandInput[k])
+    )
+  }
+
+  async run() {
+    this.quest = await Quest.findById(this.session.quest)
+    const stages = this.getStages()
+    if (stages.length === 0) {
+      console.error('stage not found', this.session.tasks)
+      return {}
+    }
+
+    this.handleEvent()
+    this.handleCommand()
+
+    const response = stages
+      .filter((s) => this.isMatch(s))
+      .reduce((r, s) => ({ ...r, ...this.execute(s) }), {})
+
+    return {
+      end: this.session.status === 'finished',
+      ...response,
+    }
   }
 }
-
-export default commandHandler
diff --git a/server/src/game/sessionHandler.js b/server/src/game/sessionHandler.js
new file mode 100644
index 0000000..124c305
--- /dev/null
+++ b/server/src/game/sessionHandler.js
@@ -0,0 +1,62 @@
+import { pushToSession } from '../api/socket.js'
+
+export default class SessionHandler {
+  constructor(session) {
+    this.session = session
+    this.quest = session.quest
+  }
+
+  getTask(stage) {
+    return this.session.tasks.find((t) => t.id === stage.id)
+  }
+
+  isCompleted(stage) {
+    const task = this.getTask(stage)
+    return task ? task.completed : false
+  }
+
+  getStages() {
+    return this.quest.stages.filter(
+      (stage) =>
+        stage.requirements.every((r) => this.isCompleted({ id: r })) &&
+        (stage.repeatable || !this.isCompleted(stage))
+    )
+  }
+
+  getNewTasks() {
+    const stages = this.getStages()
+    return stages
+      .filter((s) => s.task)
+      .filter((s) => !this.session.tasks.some((t) => t.id === s.id))
+      .map((s) => ({ id: s.id, name: s.task, completed: false }))
+  }
+
+  addNewTasks() {
+    const newTasks = this.getNewTasks()
+    this.session.tasks.push(...newTasks)
+    pushToSession(this.session.id, 'tasks', this.session.tasks)
+  }
+
+  execute(stage) {
+    this.session.tasks
+      .filter((t) => t.id === stage.id)
+      .forEach((t) => {
+        t.completed = true
+      })
+
+    this.session.hints.push(...stage.hints)
+    pushToSession(this.session.id, 'hints', stage.hints)
+
+    if (stage.id === 'END') {
+      this.session.status = 'finished'
+      this.session.finishedAt = new Date()
+    }
+
+    this.addNewTasks()
+
+    return {
+      responses: stage.responses,
+      hints: stage.hints,
+    }
+  }
+}
diff --git a/server/src/models/quest.js b/server/src/models/quest.js
index ea592f1..84c1966 100644
--- a/server/src/models/quest.js
+++ b/server/src/models/quest.js
@@ -15,19 +15,25 @@ const stageSchema = new Schema({
     type: String,
     required: true,
   },
+  task: {
+    type: String,
+  },
+  requirements: [
+    {
+      type: String,
+      required: true,
+    },
+  ],
+  repeatable: {
+    type: Boolean,
+    required: true,
+    default: false,
+  },
   condition: {
-    command: {
-      type: [String],
-      required: false,
-    },
-    output: {
-      type: [String],
-      required: false,
-    },
-    error: {
-      type: [String],
-      required: false,
-    },
+    command: [String],
+    output: [String],
+    error: [String],
+    pwd: [String],
   },
   responses: {
     type: [responseSchema],
@@ -37,10 +43,6 @@ const stageSchema = new Schema({
     type: [String],
     required: true,
   },
-  next: {
-    type: String,
-    required: true,
-  },
 })
 
 const Quest = model(
@@ -50,6 +52,18 @@ const Quest = model(
     title: String,
     order: Number,
     instruction: String,
+    tasks: {
+      id: String,
+      content: String,
+      display: {
+        type: Boolean,
+        default: false,
+      },
+      completed: {
+        type: Boolean,
+        default: false,
+      },
+    },
     stages: [stageSchema],
   })
 )
diff --git a/server/src/models/session.js b/server/src/models/session.js
index 1ae266c..ddb4fad 100644
--- a/server/src/models/session.js
+++ b/server/src/models/session.js
@@ -1,4 +1,33 @@
 import { model, Schema } from 'mongoose'
+import { FileNode } from '@linux-odyssey/file-graph'
+
+const nodeSchema = new Schema({
+  path: String,
+  type: String,
+  discovered: {
+    type: Boolean,
+    default: false,
+  },
+})
+
+nodeSchema.add({
+  children: [nodeSchema],
+})
+
+const taskSchema = new Schema({
+  id: {
+    type: String,
+    required: true,
+  },
+  name: {
+    type: String,
+    required: true,
+  },
+  completed: {
+    type: Boolean,
+    default: false,
+  },
+})
 
 const Session = model(
   'Session',
@@ -25,9 +54,15 @@ const Session = model(
         default: Date.now,
         required: true,
       },
-      progress: String,
-      completion: [String],
+      tasks: [taskSchema],
       hints: [String],
+      graph: {
+        type: nodeSchema,
+        default: new FileNode({
+          path: '/',
+          type: 'folder',
+        }),
+      },
     },
     { timestamps: true }
   )
diff --git a/server/src/utils/auth.js b/server/src/utils/auth.js
index d97de3a..49d3ac6 100644
--- a/server/src/utils/auth.js
+++ b/server/src/utils/auth.js
@@ -19,7 +19,7 @@ export function genSessionJWT(session) {
     jwt.sign(
       { session_id: session.id },
       config.jwtSecret,
-      { expiresIn: '1h' },
+      { expiresIn: config.expiry },
       (err, token) => {
         if (err) {
           reject(err)
diff --git a/server/src/utils/quest.js b/server/src/utils/quest.js
index e8a2276..315528f 100644
--- a/server/src/utils/quest.js
+++ b/server/src/utils/quest.js
@@ -4,7 +4,7 @@ import yaml from 'yaml'
 
 import Quest from '../models/quest.js'
 
-const questDirectory = path.join(process.cwd(), 'quests')
+const questDirectory = path.join(process.cwd(), '..', 'quests')
 
 export default async function loadAndUpdateQuests() {
   const questNames = await fs.readdir(questDirectory)
diff --git a/server/swagger.yaml b/server/swagger.yaml
index 34f9bf1..69e65a8 100644
--- a/server/swagger.yaml
+++ b/server/swagger.yaml
@@ -392,7 +392,7 @@ components:
           type: string
           description: current stage id
           example: helloworld
-        completion:
+        completed:
           type: array
           description: list of completed stage ids
           items:
diff --git a/server/tests/api/sessionController.test.js b/server/tests/api/sessionController.test.js
deleted file mode 100644
index 991647b..0000000
--- a/server/tests/api/sessionController.test.js
+++ /dev/null
@@ -1,56 +0,0 @@
-import jest from 'jest'
-import * as sessionController from '../../src/api/controllers/sessionController'
-import Session from '../../src/models/session'
-
-jest.mock('../../src/models/session')
-
-describe('Session Controller', () => {
-  let mockReq
-  let mockRes
-
-  beforeEach(() => {
-    mockReq = {
-      user: { id: 'userid' },
-      body: {},
-      params: {},
-    }
-    mockRes = {
-      status: jest.fn().mockReturnThis(),
-      json: jest.fn().mockReturnThis(),
-      end: jest.fn().mockReturnThis(),
-    }
-  })
-
-  test('getSessionList returns a list of sessions', async () => {
-    Session.find.mockResolvedValue(['session1', 'session2'])
-    await sessionController.getSessionList(mockReq, mockRes)
-    expect(mockRes.json).toHaveBeenCalledWith(['session1', 'session2'])
-  })
-
-  test('createSession creates a new session', async () => {
-    mockReq.body.exercise_id = 'exerciseid'
-    const mockSession = { user: 'userid', exercise: 'exerciseid' }
-    Session.mockReturnValue(mockSession)
-    mockSession.save = jest.fn().mockResolvedValue('newsess')
-    await sessionController.createSession(mockReq, mockRes)
-    expect(mockRes.status).toHaveBeenCalledWith(201)
-    expect(mockRes.json).toHaveBeenCalledWith('newsess')
-  })
-
-  test('getSessionById returns the correct session', async () => {
-    mockReq.params.session_id = 'sessionid'
-    const mockSession = { user: 'userid', _id: 'sessionid' }
-    Session.findOne.mockResolvedValue(mockSession)
-    await sessionController.getSessionById(mockReq, mockRes)
-    expect(mockRes.json).toHaveBeenCalledWith(mockSession)
-  })
-
-  test('deleteSessionById deletes a session', async () => {
-    mockReq.params.session_id = 'sessionid'
-    const mockSession = { user: 'userid', _id: 'sessionid' }
-    Session.findOneAndDelete.mockResolvedValue(mockSession)
-    await sessionController.deleteSessionById(mockReq, mockRes)
-    expect(mockRes.status).toHaveBeenCalledWith(204)
-    expect(mockRes.end).toHaveBeenCalled()
-  })
-})
diff --git a/yarn.lock b/yarn.lock
index fc0cbb7..c3c6bed 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -1212,6 +1212,15 @@ axios@^1.4.0:
     form-data "^4.0.0"
     proxy-from-env "^1.1.0"
 
+axios@^1.5.0:
+  version "1.5.0"
+  resolved "https://registry.yarnpkg.com/axios/-/axios-1.5.0.tgz#f02e4af823e2e46a9768cfc74691fdd0517ea267"
+  integrity sha512-D4DdjDo5CY50Qms0qGQTTw6Q44jl7zRwY7bthds06pUGfChBCTcQs+N743eFWGEd6pRTMd6A+I87aWyFV5wiZQ==
+  dependencies:
+    follow-redirects "^1.15.0"
+    form-data "^4.0.0"
+    proxy-from-env "^1.1.0"
+
 babel-jest@^29.6.1:
   version "29.6.1"
   resolved "https://registry.yarnpkg.com/babel-jest/-/babel-jest-29.6.1.tgz#a7141ad1ed5ec50238f3cd36127636823111233a"