Added some basic unit test