Added simple unit test
Added simple unit test