Add tests
Code currently has no test coverage at all.
issue