Sounds good to me, as this PR is closed, would you be able to raise a PR with that addition please?