Building out some unit test stuff