add basic unit tests