Just updated the repo..let me know what you think