diff --git a/test_gpt4.py b/test_gpt4.py
index 741aa34..b24bde4 100644
--- a/test_gpt4.py
+++ b/test_gpt4.py
@@ -12,9 +12,15 @@ def test_login(self):
     def test_ask_question(self):
         question = 'Test question'
         self.ap.login()
-        self.ap.ask_question(question,10)
+        self.ap.ask_question(question,20)
         response = self.ap.get_response()
         pass
+    
+    def test_design(self):
+        question = 'A cow'
+        self.ap.login()
+        self.ap.design(question)
+        pass
 
     def tearDown(self):
         self.ap.close()