Expand unit tests #56
Description
Unit tests are recently added to the repository - adding more to improve coverage would be greatly appreciated. These are relatively simple to write, and I'm happy to review any PRs that introduce additional unit tests.