diff --git a/README.md b/README.md
index f9cbd10..1fef127 100644
--- a/README.md
+++ b/README.md
@@ -1,2 +1,8 @@
 # Matrix-Inverse-Free-WMMSE
-This is the code implementation for the Matrix-Inverse-Free  MU-MIMO WMMSE Beamforming Algorithm.
+This is the code implementation for the [Matrix-Inverse-Free MU-MIMO WMMSE Beamforming Algorithm](https://ieeexplore.ieee.org/abstract/document/10021866).
+This is a pytorch version that you can easily run with "python inverse_free.py". 
+
+## Files in this repo
+`inverse_free.py`: The main function. Can be directly run to get the results.
+
+`utils.py`: This file contains the util functions.