diff --git a/requirements.txt b/requirements.txt
index 92c0908..d1f36be 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 openai==0.27
 termcolor==2.2.0
-colorama
\ No newline at end of file
+colorama
+python-dotenv
\ No newline at end of file
diff --git a/yolo.py b/yolo.py
index bc06396..5b8ac8f 100755
--- a/yolo.py
+++ b/yolo.py
@@ -115,13 +115,12 @@ if __name__ == "__main__":
       print ("No user prompt specified.")
       sys.exit(-1)
  
-  
   # Load the correct prompt based on Shell and OS and append the user's prompt
   prompt = get_full_prompt(user_prompt, shell)
 
   # Make the first line also the system prompt
-  system_prompt = prompt[1]
-
+  system_prompt = prompt.split("\n")[0]
+  
   # Call the ChatGPT API
   response = openai.ChatCompletion.create(
     model="gpt-3.5-turbo",