diff --git a/README.md b/README.md
index f263d9c..7456aff 100644
--- a/README.md
+++ b/README.md
@@ -6,6 +6,8 @@ Implementation of <a href="https://arxiv.org/abs/2006.11239">Denoising Diffusion
 
 <img src="./sample.png" width="500px"><img>
 
+[![PyPI version](https://badge.fury.io/py/denoising-diffusion-pytorch.svg)](https://badge.fury.io/py/denoising-diffusion-pytorch)
+
 ## Install
 
 ```bash
@@ -73,8 +75,6 @@ trainer = Trainer(
 trainer.train()
 ```
 
-Todo: Command line tool for one-line training
-
 ## Citations
 
 ```bibtex