diff --git a/README.md b/README.md
index 4e421b8d558fa163cfb93ea5b64efde2f7148e2b..437258ae0f6d7c7f2c3ae2ed5f7079e59fb89d6d 100644
--- a/README.md
+++ b/README.md
@@ -8,11 +8,15 @@ This repository contains supporting code for the paper:
 
     Improving the security of LCB block cipher against Deep Learning-based attacks
 
-There are 4 seperate folders in this repository:
+There are 5 seperate folders in this repository:
 
     LBC-IoT_Github - contains files used to train the deep learning model as a distinguisher for LBC-IoT
     SLIM_Github - contains files used to train the deep learning model as a distinguisher for SLIM
     LBC-IoT-Key-Recovery - contains files to run a key ranking attack on LBC-IoT
-    NIST - contains file to run a data generation code for Secure LCB for analysis of NIST statistical tests.
+    NIST - contains file to run a data generation code for Secure LCB for analysis of NIST statistical tests
+    Difference - contains .py files to be used with AutoND's repository
 
 Each of the folders contains a Jupyter Notebook which can be used to run the respective programs.
+
+The input difference for the ciphers has been calculated using https://github.com/Crypto-TII/AutoND
+The deep learning model used in this work is taken from Gohr's work and some of the code are adapted from https://github.com/agohr/deep_speck