Add some unit test