added some simple unit tests