add basic unit test