added some unit test stuff