Added some unit tests.

This shores up coverage numbers.