adding a couple of simple unit tests