diff --git a/README.md b/README.md
index 453b4a438..f1c4e8409 100644
--- a/README.md
+++ b/README.md
@@ -95,7 +95,9 @@ If you're interested in contributing to the `prompt2model` project, please
 
 We have [written a paper describing Prompt2Model in detail](https://arxiv.org/abs/2308.12261).
 
-If you use Prompt2Model in your research, please cite our paper:
+If you use Prompt2Model in your research, please cite us!
+
+If you discuss or use the overall prompt2model framework, please reference
 
 ```bibtex
 @misc{prompt2model,
@@ -107,3 +109,16 @@ If you use Prompt2Model in your research, please cite our paper:
       primaryClass={cs.CL}
 }
 ```
+
+If you discuss or use our dataset retrieval and transformation tools, please reference
+
+```bibtex
+@misc{prompt2modeldatatune,
+      title={Better Synthetic Data by Retrieving and Transforming Existing Datasets}, 
+      author={Saumya Gandhi and Ritu Gala and Vijay Viswanathan and Tongshuang Wu and Graham Neubig},
+      year={2024},
+      eprint={2404.14361},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```