Update README.md

README.md

@@ -99,7 +99,7 @@ Feel free to design and test your prompts and to share your thoughts with us, wh
 ## Reproducing MedQA Performance with vLLM
 
 Here is an example code for fast model evaluation in MedQA using vLLM. To adapt this code for other datasets like MedMCQA or MMLU, simply modify the instructions and update the dataset paths as needed.
-```
+```python
 import re
 from datasets import load_dataset
 from vllm import LLM, SamplingParams
@@ -187,7 +187,7 @@ Please note that the scores in MMLU-Medical were calculated based on the average
 ## Reference
 
 Please see the information below to cite our paper.
-```
+```bibtex
 @article{kim2024small,
   title={Small language models learn enhanced reasoning skills from medical textbooks},
   author={Kim, Hyunjae and Hwang, Hyeon and Lee, Jiwoo and Park, Sihyeon and Kim, Dain and Lee, Taewhoo and Yoon, Chanwoong and Sohn, Jiwoong and Choi, Donghee and Kang, Jaewoo},