Increase test coverage

Adds a couple passing tests!