diff --git a/mitdeeplearning/lab1.py b/mitdeeplearning/lab1.py
index 19d448a3..1b826dfe 100644
--- a/mitdeeplearning/lab1.py
+++ b/mitdeeplearning/lab1.py
@@ -7,7 +7,7 @@
 
 from IPython.display import Audio
 
-
+#current working directory
 cwd = os.path.dirname(__file__)
 
 def load_training_data():
@@ -17,7 +17,7 @@ def load_training_data():
     return songs
 
 def extract_song_snippet(text):
-    pattern = '(^|\n\n)(.*?)\n\n'
+    pattern = '(^|\n\n\n)(.*?)\n\n\n'
     search_results = re.findall(pattern, text, overlapped=True, flags=re.DOTALL)
     songs = [song[1] for song in search_results]
     print("Found {} songs in text".format(len(songs)))