As per suggestion in #116 . Adding a patch . Please review.