diff --git a/README.md b/README.md
index 6c58aee..cd17aa3 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,5 @@
-# Conditional Neural Movement Primitives
+# Conditional Neural Movement Primitives 
+**This branch includes Keras/Tensorflow implementation. For PyTorch implementation please refer to [master](https://github.com/myunusseker/CNMP/tree/master) branch**
 
 ![CNMP.png](CNMP.png)