diff --git a/README.md b/README.md
index fe350eb..27be619 100644
--- a/README.md
+++ b/README.md
@@ -18,7 +18,7 @@ The following EEG datasets were used in the project:
 ### Dependencies
 
 * Python 3
-* EEGExtract.py
+* [EEGExtract](https://github.com/sari-saba-sadiya/EEGExtract).
 * Scikit-learn
 * Numpy
 * Pandas
diff --git a/eeg_ml_pipeline/EEGExtract.py b/eeg_ml_pipeline/EEGExtract.py
index 88c5236..109b65b 100644
--- a/eeg_ml_pipeline/EEGExtract.py
+++ b/eeg_ml_pipeline/EEGExtract.py
@@ -3,7 +3,10 @@
 
 # In[ ]:
 
-
+##################################################
+# See original library for updates
+# https://github.com/sari-saba-sadiya/EEGExtract
+##################################################
 import bisect
 import numpy as np
 import pandas as pd