diff --git a/src/main/java/Tools/MaxKb/Test.java b/src/main/java/Tools/MaxKb/Test.java
index cc1caad6..d2b78aec 100644
--- a/src/main/java/Tools/MaxKb/Test.java
+++ b/src/main/java/Tools/MaxKb/Test.java
@@ -1,11 +1,9 @@
 package Tools.MaxKb;
 
 import Tools.MaxKb.Util.MaxKbDb;
-import com.dsideal.QingLong.Util.CommonUtil;
 import com.jfinal.kit.PropKit;
 import com.jfinal.plugin.activerecord.Db;
 import com.jfinal.plugin.activerecord.Record;
-import org.joda.time.DateTime;
 
 import java.util.UUID;
 
@@ -35,12 +33,17 @@ public class Test {
 
     /**
      * 增加应用程序
+     *
      * @param sourceApplicationName 源应用名称
      * @param targetApplicationName 目标应用名称
      */
-    public static void addApplication(String sourceApplicationName,String targetApplicationName){
+    public static void addApplication(String sourceApplicationName, String targetApplicationName) {
         //获取应用
         Record record = getApplication(sourceApplicationName);
+        if (record == null) {
+            System.out.println(sourceApplicationName + "应用不存在！");
+            System.exit(0);
+        }
         //克隆出对象
         record = new Record().setColumns(record.getColumns());
         //生成一个uuid
@@ -51,6 +54,41 @@ public class Test {
         Db.save("application", "id", record);
     }
 
+    /**
+     * 获取数据集
+     *
+     * @param dataset_name 数据集名称
+     * @return
+     */
+    public static Record getDataSet(String dataset_name) {
+        String sql = "select * from dataset where name=?";
+        return Db.findFirst(sql, dataset_name);
+    }
+
+    /**
+     * 增加数据集
+     *
+     * @param sourceDatasetName 数据集名称
+     */
+    public static void addDataSet(String sourceDatasetName, String targetDatasetName) {
+        //获取数据集
+        Record record = getDataSet(sourceDatasetName);
+        if (record == null) {
+            System.out.println(sourceDatasetName + "数据集不存在！");
+            System.exit(0);
+        }
+        //克隆出对象
+        record = new Record().setColumns(record.getColumns());
+        //生成一个uuid
+        record.set("id", UUID.randomUUID());
+        //名称
+        record.set("name", targetDatasetName);
+        record.set("desc", targetDatasetName);
+        //增加
+        Db.save("dataset", "id", record);
+    }
+
+
     public static void main(String[] args) throws Exception {
         //加载配置文件
         PropKit.use("MaxKb.properties");
@@ -65,10 +103,13 @@ public class Test {
 
         //增加应用
         String sourceApplicationName = "黄海的一个应用";
-        String targetApplicationName="XXX的拷贝应用";
-        addApplication(sourceApplicationName,targetApplicationName);
+        String targetApplicationName = "XXX的拷贝应用";
+        addApplication(sourceApplicationName, targetApplicationName);
         System.out.println("增加成功！");
 
         //每个人一个应用，每个人一个知识库
+        String sourceDataSetName = "长春教育知识库";
+        String targetDataSetName = "XXX的拷贝知识库";
+        addDataSet(sourceDataSetName, targetDataSetName);
     }
 }
\ No newline at end of file