diff --git a/README.md b/README.md
index 60945dc..5447b04 100644
--- a/README.md
+++ b/README.md
@@ -49,15 +49,9 @@ pip install -r assets/requirements/requirements.txt
 BiomedParseData was created from preprocessing publicly available biomedical image segmentation datasets. Check a subset of our processed datasets on HuggingFace: https://huggingface.co/datasets/microsoft/BiomedParseData. For the source datasets, please check the details here: [BiomedParseData](assets/readmes/DATASET.md). As a quick start, we've samples a tiny demo dataset at biomedparse_datasets/BiomedParseData-Demo
 
 ## Model Checkpoints
-We host our model checkpoints on HuggingFace here: https://huggingface.co/microsoft/BiomedParse.
+We host our model checkpoints on HuggingFace here: https://huggingface.co/microsoft/BiomedParse. See example code below on model loading.
 
-Step 1. Create pretrained model folder
-```
-mkdir pretrained
-```
-Step 2. Download model checkpoint and put the model in the pretrained folder when running the code. Change file name to biomed_parse.pt
-
-Expect future updates of the model as we are making it more robust and powerful based on feedbacks from the community. We recomment using the latest version of the model.
+Please expect future updates of the model as we are making it more robust and powerful based on feedbacks from the community. We recomment using the latest version of the model.
 
 ## Running Inference with BiomedParse
 
@@ -74,7 +68,9 @@ To perform inference with BiomedParse, use the provided example code and resourc
 
 We’ve included sample notebooks to guide you through running inference with BiomedParse:
 
+- **RGB Inference Example**: Check out the `inference_examples_RGB.ipynb` notebook for example using normal RGB images, including Pathology, X-ray, Ultrasound, Endoscopy, Dermoscopy, OCT, Fundus.
 - **DICOM Inference Example**: Check out the `inference_examples_DICOM.ipynb` notebook for example using DICOM images.
+- **NIFTI Inference Example**: Check out the `inference_examples_NIFTI.ipynb` notebook for example using NIFTI image slices.
 - You can also try a quick online demo: [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/microsoft/BiomedParse/blob/main/inference_colab_demo.ipynb)
 
 ### Model Setup
@@ -87,6 +83,7 @@ from utilities.distributed import init_distributed
 from utilities.arguments import load_opt_from_config_files
 from utilities.constants import BIOMED_CLASSES
 from inference_utils.inference import interactive_infer_image
+from inference_utils.output_processing import check_mask_stats
 import numpy as np
 
 # Build model config
@@ -124,12 +121,14 @@ for i, pred in enumerate(pred_mask):
     gt = gt_masks[i]
     dice = (1*(pred>0.5) & gt).sum() * 2.0 / (1*(pred>0.5).sum() + gt.sum())
     print(f'Dice score for {prompts[i]}: {dice:.4f}')
+    check_mask_stats(image, pred_mask[i]*255, 'X-Ray-Chest', text_prompt[i])
+    print(f'p-value for {prompts[i]}: {p_value:.4f}')
 ```
 
 
 Detection and recognition inference code are provided in `inference_utils/output_processing.py`.
 
-- `check_mask_stats()`: Outputs p-value for model-predicted mask for detection.
+- `check_mask_stats()`: Outputs p-value for model-predicted mask for detection. Check the `inference_examples_RGB.ipynb` notebook.
 - `combine_masks()`: Combines predictions for non-overlapping masks.
 
 ## Finetune on Your Own Data
diff --git a/assets/scripts/eval.sh b/assets/scripts/eval.sh
index 74a6c48..4aade4f 100755
--- a/assets/scripts/eval.sh
+++ b/assets/scripts/eval.sh
@@ -17,5 +17,5 @@ CUDA_VISIBLE_DEVICES=0 mpirun -n 1 python entry.py evaluate \
             FP16 True \
             WEIGHT True \
             STANDARD_TEXT_FOR_EVAL False \
-            RESUME_FROM pretrained/biomed_parse.pt \
+            RESUME_FROM pretrained/biomedparse_v1.pt \
             
\ No newline at end of file
diff --git a/assets/scripts/train.sh b/assets/scripts/train.sh
index 407223f..ea74d99 100755
--- a/assets/scripts/train.sh
+++ b/assets/scripts/train.sh
@@ -37,4 +37,4 @@ CUDA_VISIBLE_DEVICES=0 mpirun -n 1 python entry.py train \
             ATTENTION_ARCH.QUERY_NUMBER 3 \
             STROKE_SAMPLER.MAX_CANDIDATE 10 \
             WEIGHT True \
-            RESUME_FROM pretrained/biomed_parse.pt
\ No newline at end of file
+            RESUME_FROM pretrained/biomedparse_v1.pt
\ No newline at end of file
diff --git a/example_prediction.py b/example_prediction.py
index 82e8ff3..2e4ab51 100644
--- a/example_prediction.py
+++ b/example_prediction.py
@@ -8,6 +8,7 @@
 import numpy as np
 
 from inference_utils.inference import interactive_infer_image
+from inference_utils.output_processing import check_mask_stats
 
 opt = load_opt_from_config_files(["configs/biomedparse_inference.yaml"])
 opt = init_distributed(opt)
@@ -41,4 +42,6 @@
 for i, pred in enumerate(pred_mask):
     gt = gt_masks[i]
     dice = (1*(pred>0.5) & gt).sum() * 2.0 / (1*(pred>0.5).sum() + gt.sum())
-    print(f'Dice score for {prompts[i]}: {dice:.4f}')
\ No newline at end of file
+    print(f'Dice score for {prompts[i]}: {dice:.4f}')
+    p_value = check_mask_stats(np.array(image), pred*255, 'Pathology', prompts[i])
+    print(f'p-value for {prompts[i]}: {p_value:.4f}')
\ No newline at end of file
diff --git a/inference_utils/output_processing.py b/inference_utils/output_processing.py
index 1472d94..17ee3ef 100644
--- a/inference_utils/output_processing.py
+++ b/inference_utils/output_processing.py
@@ -12,6 +12,7 @@ def check_mask_stats(img, mask, modality_type, target):
     # target: str, see target_dist.json for the list of targets
     
     huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename='target_dist.json', local_dir='./inference_utils')
+    huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename="config.yaml", local_dir="./configs")
     target_dist = json.load(open("inference_utils/target_dist.json"))
     
     if modality_type not in target_dist: