Improve API

Signed-off-by: Aniket Pant <me@aniketpant.com>
Add documentation on usage

Signed-off-by: Aniket Pant <me@aniketpant.com>