diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index 5325cb1..7e3ed0a 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -11,7 +11,7 @@ on:
       - main
 jobs:
   lint:
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
     steps:
       - name: Checkout
         uses: actions/checkout@v3
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 9eeab8e..8d7be22 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -10,7 +10,7 @@ on:
 
 jobs:
   testing:
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python
diff --git a/README.md b/README.md
index 80fe5b8..b4f032f 100644
--- a/README.md
+++ b/README.md
@@ -212,6 +212,17 @@ Please check the [CONTRIBUTING.md](CONTRIBUTING.md) file if you want to contribu
 
 [![Star History Chart](https://api.star-history.com/svg?repos=MLSysOps/MLE-agent&type=Date)](https://star-history.com/#MLSysOps/MLE-agent&Date)
 
+## Citation
+
+```bibtex
+@misc{zhang2024mleagent,
+title = {MLE-Agent: Your Intelligent Companion for Seamless AI Engineering and Research},
+author = {Zhang, Huaizheng and Huang, Yizheng},
+year = {2024},
+note = {\url{https://github.com/MLSysOps/MLE-agent}},
+}
+```
+
 ## License
 
 Check [MIT License](LICENSE) file for more information.