Update README.md

README.md

@@ -78,4 +78,20 @@ corrupted_programs = corrupt(
 for corrupted_program in corrupted_programs:
     print('-'*30)
     print(corrupted_program)
+```
+
+# Citation
+
+If you use our benchmark or bug injection model, please cite our paper.
+
+```
+@misc{szalontai2026diffbasedcodecorruptionusing,
+      title={Diff-Based Code Corruption using LLMs for Large-Scale Bugfix Benchmarking}, 
+      author={Balázs Szalontai and Ábel Szauter and Balázs Márton and Péter Verebics and Balázs Pintér and Tibor Gregorics},
+      year={2026},
+      eprint={2606.29088},
+      archivePrefix={arXiv},
+      primaryClass={cs.SE},
+      url={https://arxiv.org/abs/2606.29088}, 
+}
 ```