diff --git a/src/Main.java b/src/Main.java
index 374dc2d..2cc2368 100644
--- a/src/Main.java
+++ b/src/Main.java
@@ -1,9 +1,11 @@
-public class Main {
-  public static void main(String[] args) {
-    ExamSystem system = new ExamSystem();
-    while (true) {
-      Users users = system.login();
-      system.generateProblems(users);
-    }
-  }
+public class Main {
+  public static void main(String[] args) {
+    ExamSystem system = new ExamSystem();
+    while (true) {
+      Users users = system.login();
+      System.out.print("准备生成" + users.getType()
+          + "数学题目，请输入生成题目数量（输入-1将退出当前用户，重新登录）：");
+      system.generateProblems(users);
+    }
+  }
 }
\ No newline at end of file