diff --git a/README.md b/README.md
index f4b2480b429875361b11173a20506d4a4903e4f4..a79fc792b2c30c0235419d0e89a06f774e04d13a 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,8 @@
 # COMP3217 code and result
 
+The source code and code run results for Assignment Part A and Part B are shown in AI_Part1.ipynb and AI_Part2.ipynb respectively.
 
+The test_pre1.csv and test_pre2.csv are the model prediction data and results of the code output. The first 128 columns are the test data and the last column (column 129) is the label of the algorithm prediction.
 
 ## Getting started