Add additional tests to increase code coverage to 100%

Fixes #11