diff --git a/README.md b/README.md
index 92ca018..8e19c7c 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # DiffJPEG: A Jax Implementation
 
-This is a Jax implementation of the differentiable JPEG compression algorithm, based on the [PyTorch implementation](https://github.com/mlomnitz/DiffJPEG).
+This is a Jax implementation of the differentiable JPEG compression algorithm, based on the [PyTorch implementation](https://github.com/mlomnitz/DiffJPEG) and some of the modifications found in [this repository](https://github.com/necla-ml/Diff-JPEG) to improve quality at high compression rates.
 
 
 ## Requirements
@@ -17,7 +17,7 @@ pip install diffjpeg_jax
 
 ## Usage
 
-Unlike the PyTorch version, this is ML library agnostic, so it simply is implemented as a function.
+Unlike the PyTorch version, this is ML library agnostic, so it simply is implemented as a function. Inputs should be in the range `[0, 255]` and in the format `(H, W, C)`.
 
 ```python