What do you think add a test and update the docs to be easier to see how it works?