add a couple unit tests