Update README.md

README.md

@@ -37,4 +37,9 @@ Below is a question and it's corresponding code answer. Please write test cases
 {code in function format}
 
 Please add detailed comments to the test cases you write. You do not need to test the function's ability to throw exceptions.
+```
+
+# Citation
+If you find our model helpful, please cite the original paper:
+```
 ```