add basic unit testing