I've managed to get this done, will be opening a pull request soon agains the github repo.
Thanks sundeep kahlon !