Add tests

This brings test coverage to 100%.