diff --git a/pyproject.toml b/pyproject.toml
index 1aede35..7ba1b38 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "nebulagraph-lite"
-version = "0.0.5"
+version = "0.0.6"
 description = "Plug and play NebulaGraph with pip install."
 authors = [
     {name = "Wey Gu",email = "weyl.gu@gmail.com"},
diff --git a/src/nebulagraph_lite/__init__.py b/src/nebulagraph_lite/__init__.py
index 063bf0a..cf0f725 100644
--- a/src/nebulagraph_lite/__init__.py
+++ b/src/nebulagraph_lite/__init__.py
@@ -2,6 +2,8 @@
 import subprocess
 import time
 import functools
+
+from urllib.request import urlretrieve
 from typing import List, Type
 
 LOCALHOST_V4 = "127.0.0.1"
@@ -9,6 +11,9 @@
 BASE_PATH = os.path.expanduser("~/.nebulagraph/lite")
 COLAB_BASE_PATH = "/content/.nebulagraph/lite"
 
+# Data set
+BASKETBALLPLAYER_DATASET_URL = "https://raw.githubusercontent.com/vesoft-inc/nebula-console/master/data/basketballplayer.ngql"
+
 
 def retry(
     exceptions: List[Type[Exception]],
@@ -122,6 +127,7 @@ def clean_up_base_path(self):
 
     def create_nebulagraph_lite_folders(self):
         try:
+            os.makedirs(os.path.join(self.base_path, "data_set"), exist_ok=True)
             os.makedirs(os.path.join(self.base_path, "data/meta0"), exist_ok=True)
             os.makedirs(os.path.join(self.base_path, "logs/meta0"), exist_ok=True)
             os.makedirs(
@@ -250,8 +256,17 @@ def activate_storaged(self):
         # TODO: do 'SHOW HOSTS' to check if storaged is activated
 
     def load_basketballplayer_dataset(self):
+        url = BASKETBALLPLAYER_DATASET_URL
+        try:
+            urlretrieve(url, f"{self.base_path}/data_set/basketballplayer.ngql")
+        except Exception as e:
+            print(e)
+            raise Exception(
+                f"Failed to download basketballplayer dataset from {url}"
+            )
+
         udocker_command = (
-            f"run --rm "
+            f"run --rm -v {self.base_path}/data_set:/root/data "
             f"vesoft/nebula-console:v3 "
             f"-addr {self.host} -port {self.port} -u root -p nebula -e ':play basketballplayer'"
         )