Tests

Adds some test coverage.