I will provide a PR for this function soon if you don't mind. @chrvadala maybe you can add me as a collaborator so I can create one :)