Actually the example in the docs shows using `vars()`, I don't mind updating to use that instead.