I have made amendments to the code in order to make the use case of this feature more apparent. Does this represent more of what you expected from the test?