Readme updated

Signed-off-by: Aniket Pant <me@aniketpant.com>
Fixed bugs and improved code

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