add some basic unit testing