added some basic unit test