Added some initial unit test