add simple unit test