diff --git a/colorization/README.md b/colorization/README.md
index 8cd66bd53bd1f8c9d7fa0212e3b77c58bc58993c..5240e70032357cd7b4126227a00bc661c05fea6d 100644
--- a/colorization/README.md
+++ b/colorization/README.md
@@ -1,6 +1,6 @@
 # Colorization
 
-See the Jupyter Notebook "Image_Colorization.ipynb" for general usage info.
+See the Jupyter Notebook "Image_Colorization.ipynb" for general usage info. The main colorization algorithm is based on https://github.com/richzhang/colorization. 
 
 ## Download training images